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 CHANGED
@@ -1,4 +1,4 @@
1
- from langchain_community.memory import ConversationBufferWindowMemory
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 langchain_community.memory import ConversationBufferWindowMemory
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 langchain_community.tools import BaseTool
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 langchain_community.tools import BaseTool
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 langchain_community.tools import BaseTool
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