Spaces:
Sleeping
Sleeping
Priyansh Saxena commited on
Commit ·
a68baeb
1
Parent(s): 1691b3a
Fix langchain imports: use langchain.memory for ConversationBufferWindowMemory, use langchain_core.tools for BaseTool (langchain 0.3.27 compatibility)
Browse files- src/agent/memory_manager.py +1 -1
- src/agent/research_agent.py +1 -1
- src/tools/base_tool.py +1 -1
- src/tools/chart_creator_tool.py +1 -1
- src/tools/chart_data_tool.py +1 -1
src/agent/memory_manager.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 2 |
from typing import Dict, Any, List, Optional
|
| 3 |
|
| 4 |
class MemoryManager:
|
|
|
|
| 1 |
+
from langchain.memory import ConversationBufferWindowMemory
|
| 2 |
from typing import Dict, Any, List, Optional
|
| 3 |
|
| 4 |
class MemoryManager:
|
src/agent/research_agent.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 2 |
from langchain_community.llms import Ollama
|
| 3 |
-
from
|
| 4 |
from typing import List, Dict, Any
|
| 5 |
import asyncio
|
| 6 |
from datetime import datetime
|
|
|
|
| 1 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 2 |
from langchain_community.llms import Ollama
|
| 3 |
+
from langchain.memory import ConversationBufferWindowMemory
|
| 4 |
from typing import List, Dict, Any
|
| 5 |
import asyncio
|
| 6 |
from datetime import datetime
|
src/tools/base_tool.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from abc import ABC, abstractmethod
|
| 2 |
from typing import Dict, Any, Optional, Union
|
| 3 |
-
from
|
| 4 |
from pydantic import BaseModel, Field, PrivateAttr, field_validator
|
| 5 |
import asyncio
|
| 6 |
import aiohttp
|
|
|
|
| 1 |
from abc import ABC, abstractmethod
|
| 2 |
from typing import Dict, Any, Optional, Union
|
| 3 |
+
from langchain_core.tools import BaseTool
|
| 4 |
from pydantic import BaseModel, Field, PrivateAttr, field_validator
|
| 5 |
import asyncio
|
| 6 |
import aiohttp
|
src/tools/chart_creator_tool.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 2 |
from pydantic import BaseModel, Field
|
| 3 |
from typing import Dict, Any, List, Optional
|
| 4 |
import json
|
|
|
|
| 1 |
+
from langchain_core.tools import BaseTool
|
| 2 |
from pydantic import BaseModel, Field
|
| 3 |
from typing import Dict, Any, List, Optional
|
| 4 |
import json
|
src/tools/chart_data_tool.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 2 |
from pydantic import BaseModel, Field
|
| 3 |
from typing import Dict, Any, List, Optional
|
| 4 |
import json
|
|
|
|
| 1 |
+
from langchain_core.tools import BaseTool
|
| 2 |
from pydantic import BaseModel, Field
|
| 3 |
from typing import Dict, Any, List, Optional
|
| 4 |
import json
|