use myscalekb agent base package

myscale · Dec 25, 2024 · a7912eb · a7912eb
1 parent d870e98
commit a7912eb
Show file tree

Hide file tree

Showing 8 changed files with 14 additions and 13 deletions.
diff --git a/myscalekb_agent_plugin/paper_recommendation/agent.py b/myscalekb_agent_plugin/paper_recommendation/agent.py
@@ -4,11 +4,11 @@
 from langchain.agents.output_parsers.tools import ToolAgentAction
 from langchain_core.runnables import RunnableConfig
 
-from myscalekb_agent.base.control_tags import ControlTags
-from myscalekb_agent.base.graph_builder import node, conditional_edge, entry, GraphBuilder, edge
-from myscalekb_agent.base.state import AgentState
-from myscalekb_agent.base.sub_agent import SubAgent
-from myscalekb_agent.base.utils import JsonHelper
+from myscalekb_agent_base.control_tags import ControlTags
+from myscalekb_agent_base.graph_builder import node, conditional_edge, entry, GraphBuilder, edge
+from myscalekb_agent_base.state import AgentState
+from myscalekb_agent_base.sub_agent import SubAgent
+from myscalekb_agent_base.utils import JsonHelper
 from myscalekb_agent_plugin.paper_recommendation.prompt import PaperRecommendationPrompt
 from myscalekb_agent_plugin.paper_recommendation.retrievers import TitleRetriever, TopicRetriever
 from myscalekb_agent_plugin.paper_recommendation.tools import SimilarityRecommendation, TopicBasedRecommendation

diff --git a/myscalekb_agent_plugin/paper_recommendation/prompt.py b/myscalekb_agent_plugin/paper_recommendation/prompt.py
@@ -1,7 +1,7 @@
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.prompts import ChatPromptTemplate
 
-from myscalekb_agent.base.prompt import Prompt
+from myscalekb_agent_base.prompt import Prompt
 
 
 class PaperRecommendationPrompt(Prompt):

diff --git a/myscalekb_agent_plugin/paper_recommendation/retrievers.py b/myscalekb_agent_plugin/paper_recommendation/retrievers.py
@@ -1,8 +1,8 @@
 import asyncio
 from typing import List, Optional, Callable, Any
 
-from myscalekb_agent.base.myscale_query import MyScaleQuery
-from myscalekb_agent.base.retriever import Retriever
+from myscalekb_agent_base.myscale_query import MyScaleQuery
+from myscalekb_agent_base.retriever import Retriever
 
 
 class TitleRetriever(Retriever):

diff --git a/myscalekb_agent_plugin/paper_recommendation/tools.py b/myscalekb_agent_plugin/paper_recommendation/tools.py
@@ -2,7 +2,7 @@
 
 from pydantic import BaseModel, Field
 
-from myscalekb_agent.base.tool import BaseTool
+from myscalekb_agent_base.tool import BaseTool
 
 
 class SimilarityRecommendation(BaseTool):

diff --git a/myscalekb_agent_plugin/simple/agent.py b/myscalekb_agent_plugin/simple/agent.py
@@ -1,5 +1,5 @@
-from myscalekb_agent.base.graph_builder import node, edge
-from myscalekb_agent.base.sub_agent import SubAgent
+from myscalekb_agent_base.graph_builder import node, edge
+from myscalekb_agent_base.sub_agent import SubAgent
 
 
 class SimpleRag(SubAgent):

diff --git a/myscalekb_agent_plugin/simple/prompt.py b/myscalekb_agent_plugin/simple/prompt.py
@@ -1,6 +1,6 @@
 from langchain_core.messages import SystemMessage
 
-from myscalekb_agent.base.prompt import Prompt
+from myscalekb_agent_base.prompt import Prompt
 
 
 class SimpleRagPrompt(Prompt):

diff --git a/myscalekb_agent_plugin/simple/retriever.py b/myscalekb_agent_plugin/simple/retriever.py
@@ -1,6 +1,6 @@
 from typing import List, Optional, Callable, Any
 
-from myscalekb_agent.base.retriever import Retriever
+from myscalekb_agent_base.retriever import Retriever
 
 
 class SimpleRagRetriever(Retriever):

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+myscalekb-agent-base==0.1.2