budivoy commited on
Commit
52f45b2
·
verified ·
1 Parent(s): cf842b1

Remove old import that shaded langchain

Browse files
Files changed (1) hide show
  1. agents.py +1 -9
agents.py CHANGED
@@ -6,7 +6,7 @@ from langgraph.prebuilt import tools_condition
6
  from langgraph.prebuilt import ToolNode
7
  from langchain_google_genai import ChatGoogleGenerativeAI
8
  from langchain_groq import ChatGroq
9
- from langchain.llms import RWKV
10
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
11
  from langchain_community.tools.tavily_search import TavilySearchResults
12
  from langchain_community.document_loaders import WikipediaLoader
@@ -16,14 +16,6 @@ from langchain_core.messages import SystemMessage, HumanMessage
16
  from langchain_core.tools import tool
17
  from huggingface_hub import hf_hub_download
18
 
19
- # RWKV setup flags — must come before importing RWKV
20
- os.environ["RWKV_JIT_ON"] = "1"
21
- os.environ["RWKV_V7_ON"] = "1" # enable RWKV-7
22
- os.environ["RWKV_CUDA_ON"] = "0" # set to "1"
23
-
24
- from rwkv.model import RWKV
25
- from rwkv.utils import PIPELINE
26
-
27
  load_dotenv()
28
 
29
  class BasicAgent:
 
6
  from langgraph.prebuilt import ToolNode
7
  from langchain_google_genai import ChatGoogleGenerativeAI
8
  from langchain_groq import ChatGroq
9
+ from langchain_community.llms import RWKV
10
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
11
  from langchain_community.tools.tavily_search import TavilySearchResults
12
  from langchain_community.document_loaders import WikipediaLoader
 
16
  from langchain_core.tools import tool
17
  from huggingface_hub import hf_hub_download
18
 
 
 
 
 
 
 
 
 
19
  load_dotenv()
20
 
21
  class BasicAgent: