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

Fix linting and type checking issues

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-04 19:53:24 +00:00
parent 583ac5711f
commit 5e29ac5f7b
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/memory/contextual/contextual_memory.py
View File

@@ -58,7 +58,7 @@ class ContextualMemory:
Fetches historical data or insights from LTM that are relevant to the task's description and expected_output,
formatted as bullet points.
"""
ltm_results = self.ltm.search(task, latest_n=2)
ltm_results = self.ltm.search(query=task, limit=2)
if not ltm_results:
return None

11
tests/memory/test_custom_storage.py
View File

@@ -1,12 +1,13 @@
import pytest
from typing import Any, Dict, List
from crewai.crew import Crew
import pytest
from crewai.agent import Agent
from crewai.memory.storage.interface import Storage
from crewai.memory.short_term.short_term_memory import ShortTermMemory
from crewai.memory.long_term.long_term_memory import LongTermMemory
from crewai.crew import Crew
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.memory.storage.interface import Storage
from crewai.memory.user.user_memory import UserMemory