andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Brandon/general cleanup (#2059)

Browse Source 
* clean up. fix type safety. address memory config docs

* improve manager

* Include fix for o1 models not supporting system messages

* more broad with o1

* address fix: Typo in expected_output string #2045

* drop prints

* drop prints

* wip

* wip

* fix failing memory tests

* Fix memory provider issue

* clean up short term memory

* revert ltm

* drop

* clean up linting issues

* more linting
This commit is contained in:
Brandon Hancock (bhancock_ai)
2025-02-07 17:00:41 -05:00
committed by GitHub
parent 74a1de8550
commit 6f4ad532e6
5 changed files with 3 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/crewai/agent.py
View File

@@ -16,7 +16,6 @@ from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.task import Task from crewai.task import Task
from crewai.tools import BaseTool from crewai.tools import BaseTool
from crewai.tools.agent_tools.agent_tools import AgentTools from crewai.tools.agent_tools.agent_tools import AgentTools
from crewai.tools.base_tool import Tool
from crewai.utilities import Converter, Prompts from crewai.utilities import Converter, Prompts
from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_FILE from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_FILE
from crewai.utilities.converter import generate_model_description from crewai.utilities.converter import generate_model_description
@@ -146,7 +145,7 @@ class Agent(BaseAgent):
def _set_knowledge(self): def _set_knowledge(self):
try: try:
if self.knowledge_sources: if self.knowledge_sources:
full_pattern = re.compile(r'[^a-zA-Z0-9\-_\r\n]|(\.\.)') full_pattern = re.compile(r"[^a-zA-Z0-9\-_\r\n]|(\.\.)")
knowledge_agent_name = f"{re.sub(full_pattern, '_', self.role)}" knowledge_agent_name = f"{re.sub(full_pattern, '_', self.role)}"
if isinstance(self.knowledge_sources, list) and all( if isinstance(self.knowledge_sources, list) and all(
isinstance(k, BaseKnowledgeSource) for k in self.knowledge_sources isinstance(k, BaseKnowledgeSource) for k in self.knowledge_sources

5
src/crewai/cli/reset_memories_command.py
View File

@@ -3,11 +3,6 @@ import subprocess
import click import click
from crewai.cli.utils import get_crew from crewai.cli.utils import get_crew
from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
from crewai.memory.entity.entity_memory import EntityMemory
from crewai.memory.long_term.long_term_memory import LongTermMemory
from crewai.memory.short_term.short_term_memory import ShortTermMemory
from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
def reset_memories_command( def reset_memories_command(

2
src/crewai/crew.py
View File

@@ -1,7 +1,6 @@
import asyncio import asyncio
import json import json
import re import re
import sys
import uuid import uuid
import warnings import warnings
from concurrent.futures import Future from concurrent.futures import Future
@@ -440,7 +439,6 @@ class Crew(BaseModel):
) )
return self return self
@property @property
def key(self) -> str: def key(self) -> str:
source = [agent.key for agent in self.agents] + [ source = [agent.key for agent in self.agents] + [

2
src/crewai/memory/entity/entity_memory.py
View File

@@ -1,4 +1,4 @@
from typing import Any, Optional from typing import Optional
from pydantic import PrivateAttr from pydantic import PrivateAttr

4
src/crewai/memory/memory.py
View File

@@ -1,9 +1,7 @@
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
from crewai.memory.storage.rag_storage import RAGStorage
class Memory(BaseModel): class Memory(BaseModel):
""" """