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:
|