Remove old import that shaded langchain
Browse files
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
|
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:
|