bisheng-langchain 0.4.0.dev1__py3-none-any.whl → 0.4.1__py3-none-any.whl

bisheng_langchain/gpts/agent_types/llm_functions_agent.py

@@ -62,7 +62,10 @@ def get_openai_functions_agent_executor(tools: list[BaseTool], llm: LanguageMode
         for tool_call in last_message.additional_kwargs['tool_calls']:
             function = tool_call['function']
             function_name = function['name']
-            _tool_input = json.loads(function['arguments'] or '{}')
+            try:
+                _tool_input = json.loads(function['arguments'] or '{}')
+            except Exception as e:
+                raise Exception(f"Error parsing arguments for function: {function_name}. arguments: {function['arguments']}. error: {str(e)}")
             # We construct an ToolInvocation from the function_call
             actions.append(ToolInvocation(
                 tool=function_name,
@@ -89,7 +92,10 @@ def get_openai_functions_agent_executor(tools: list[BaseTool], llm: LanguageMode
         for tool_call in last_message.additional_kwargs['tool_calls']:
             function = tool_call['function']
             function_name = function['name']
-            _tool_input = json.loads(function['arguments'] or '{}')
+            try:
+                _tool_input = json.loads(function['arguments'] or '{}')
+            except Exception as e:
+                raise Exception(f"Error parsing arguments for function: {function_name}. arguments: {function['arguments']}. error: {str(e)}")
             # We construct an ToolInvocation from the function_call
             actions.append(ToolInvocation(
                 tool=function_name,
@@ -200,7 +206,11 @@ def get_qwen_local_functions_agent_executor(
         # only one function
         function = last_message.additional_kwargs['function_call']
         function_name = function['name']
-        _tool_input = json.loads(function['arguments'] or '{}')
+        try:
+            _tool_input = json.loads(function['arguments'] or '{}')
+        except Exception as e:
+            raise Exception(
+                f"Error parsing arguments for function: {function_name}. arguments: {function['arguments']}. error: {str(e)}")
         # We construct an ToolInvocation from the function_call
         actions.append(ToolInvocation(
             tool=function_name,

bisheng_langchain/gpts/load_tools.py

@@ -6,26 +6,27 @@ from typing import Any, Callable, Dict, List, Optional, Tuple
 import httpx
 import pandas as pd
 import pymysql
+from dotenv import load_dotenv
+from langchain_community.tools.arxiv.tool import ArxivQueryRun
+from langchain_community.tools.bearly.tool import BearlyInterpreterTool
+from langchain_community.utilities.arxiv import ArxivAPIWrapper
+from langchain_community.utilities.bing_search import BingSearchAPIWrapper
+from langchain_core.callbacks import BaseCallbackManager, Callbacks
+from langchain_core.language_models import BaseLanguageModel
+from langchain_core.tools import BaseTool, Tool
+from mypy_extensions import Arg, KwArg
+
 from bisheng_langchain.gpts.tools.api_tools import ALL_API_TOOLS
 from bisheng_langchain.gpts.tools.bing_search.tool import BingSearchRun
 from bisheng_langchain.gpts.tools.calculator.tool import calculator
 from bisheng_langchain.gpts.tools.code_interpreter.tool import CodeInterpreterTool
-
 # from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
 from bisheng_langchain.gpts.tools.dalle_image_generator.tool import (
     DallEAPIWrapper,
     DallEImageGenerator,
 )
 from bisheng_langchain.gpts.tools.get_current_time.tool import get_current_time
-from dotenv import load_dotenv
-from langchain_community.tools.arxiv.tool import ArxivQueryRun
-from langchain_community.tools.bearly.tool import BearlyInterpreterTool
-from langchain_community.utilities.arxiv import ArxivAPIWrapper
-from langchain_community.utilities.bing_search import BingSearchAPIWrapper
-from langchain_core.callbacks import BaseCallbackManager, Callbacks
-from langchain_core.language_models import BaseLanguageModel
-from langchain_core.tools import BaseTool, Tool
-from mypy_extensions import Arg, KwArg
+from bisheng_langchain.gpts.tools.sql_agent.tool import SqlAgentTool, SqlAgentAPIWrapper
 from bisheng_langchain.rag import BishengRAGTool
 from bisheng_langchain.utils.azure_dalle_image_generator import AzureDallEWrapper
 
@@ -80,6 +81,10 @@ def _get_dalle_image_generator(**kwargs: Any) -> Tool:
     )
 
 
+def _get_sql_agent(**kwargs: Any) -> BaseTool:
+    return SqlAgentTool(api_wrapper=SqlAgentAPIWrapper(**kwargs))
+
+
 def _get_bearly_code_interpreter(**kwargs: Any) -> Tool:
     return BearlyInterpreterTool(**kwargs).as_tool()
 
@@ -99,6 +104,7 @@ _EXTRA_PARAM_TOOLS: Dict[str, Tuple[Callable[[KwArg(Any)], BaseTool], List[Optio
     'bisheng_rag': (BishengRAGTool.get_rag_tool, ['name', 'description'],
                     ['vector_store', 'keyword_store', 'llm', 'collection_name', 'max_content',
                      'sort_by_source_and_index']),
+    'sql_agent': (_get_sql_agent, ['llm', 'sql_address'], []),
 }
 
 _API_TOOLS: Dict[str, Tuple[Callable[[KwArg(Any)], BaseTool], List[str]]] = {**ALL_API_TOOLS}  # type: ignore

bisheng_langchain/gpts/tools/api_tools/openapi.py

@@ -9,8 +9,11 @@ from .base import APIToolBase, Field, MultArgsSchemaTool
 
 class OpenApiTools(APIToolBase):
 
-    def get_real_path(self):
-        return self.url + self.params['path']
+    def get_real_path(self, path_params: dict|None):
+        path = self.params['path']
+        if path_params:
+            path = path.format(**path_params)
+        return self.url + path
 
     def get_request_method(self):
         return self.params['method'].lower()
@@ -20,17 +23,20 @@ class OpenApiTools(APIToolBase):
         for one in self.params['parameters']:
             params_define[one['name']] = one
 
+        path_params = {}
         params = {}
         json_data = {}
         for k, v in kwargs.items():
             if params_define.get(k):
                 if params_define[k]['in'] == 'query':
                     params[k] = v
+                elif params_define[k]['in'] == 'path':
+                    path_params[k] = v
                 else:
                     json_data[k] = v
             else:
                 params[k] = v
-        return params, json_data
+        return params, json_data, path_params
 
     def parse_args_schema(self):
         params = self.params['parameters']
@@ -43,6 +49,10 @@ class OpenApiTools(APIToolBase):
                 field_type = int
             elif field_type == 'string':
                 field_type = str
+            elif field_type == 'boolean':
+                field_type = bool
+            elif field_type == 'array':
+                field_type = list
             elif field_type in {'object', 'dict'}:
                 param_object_param = {}
                 for param in one['schema']['properties'].keys():
@@ -53,6 +63,10 @@ class OpenApiTools(APIToolBase):
                         field_type = int
                     elif field_type == 'string':
                         field_type = str
+                    elif field_type == 'boolean':
+                        field_type = bool
+                    elif field_type == 'array':
+                        field_type = list
                     param_object_param[param] = (
                         field_type,
                         Field(description=one['schema']['properties'][param]['description']))
@@ -74,10 +88,10 @@ class OpenApiTools(APIToolBase):
         extra = {}
         if 'proxy' in kwargs:
             extra['proxy'] = kwargs.pop('proxy')
-        path = self.get_real_path()
+        params, json_data, path_params = self.get_params_json(**kwargs)
+        path = self.get_real_path(path_params)
         logger.info('api_call url={}', path)
         method = self.get_request_method()
-        params, json_data = self.get_params_json(**kwargs)
 
         if method == 'get':
             resp = self.client.get(path, params=params, **extra)
@@ -100,10 +114,10 @@ class OpenApiTools(APIToolBase):
         if 'proxy' in kwargs:
             extra['proxy'] = kwargs.pop('proxy')
 
-        path = self.get_real_path()
+        params, json_data, path_params = self.get_params_json(**kwargs)
+        path = self.get_real_path(path_params)
         logger.info('api_call url={}', path)
         method = self.get_request_method()
-        params, json_data = self.get_params_json(**kwargs)
 
         if method == 'get':
             resp = await self.async_client.aget(path, params=params, **extra)

bisheng_langchain/gpts/tools/sql_agent/tool.py

@@ -0,0 +1,262 @@
+from typing import Type, Optional, TypedDict, Annotated, Any, Literal
+
+from langchain_community.agent_toolkits import SQLDatabaseToolkit
+from langchain_community.utilities import SQLDatabase
+from langchain_core.callbacks import CallbackManagerForToolRun
+from langchain_core.language_models import BaseLanguageModel
+from langchain_core.messages import AnyMessage, AIMessage, ToolMessage
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.runnables import RunnableLambda, RunnableWithFallbacks
+from langchain_core.tools import BaseTool, tool
+from langgraph.constants import END, START
+from langgraph.graph import add_messages, StateGraph
+from langgraph.prebuilt import ToolNode
+from pydantic import BaseModel, Field
+
+
+class State(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+
+
+def handle_tool_error(state) -> dict:
+    error = state.get("error")
+    tool_calls = state["messages"][-1].tool_calls
+    return {
+        "messages": [
+            ToolMessage(
+                content=f"Error: {repr(error)}\n please fix your mistakes.",
+                tool_call_id=tc["id"],
+            )
+            for tc in tool_calls
+        ]
+    }
+
+
+def create_tool_node_with_fallback(tools: list) -> RunnableWithFallbacks[Any, dict]:
+    """
+    Create a ToolNode with a fallback to handle errors and surface them to the agent.
+    """
+    return ToolNode(tools).with_fallbacks(
+        [RunnableLambda(handle_tool_error)], exception_key="error"
+    )
+
+
+class SubmitFinalAnswer(BaseModel):
+    """Submit the final answer to the user based on the query results."""
+
+    final_answer: str = Field(..., description="The final answer to the user")
+
+class QueryDBTool(BaseTool):
+    name = "db_query_tool"
+    description = """Execute a SQL query against the database and get back the result.
+        If the query is not correct, an error message will be returned.
+        If an error is returned, rewrite the query, check the query, and try again."""
+
+    db: SQLDatabase
+
+    def _run(self, query: str, run_manager: Optional[CallbackManagerForToolRun] = None):
+        result = self.db.run_no_throw(query)
+        if not result:
+            return "Error: Query failed. Please rewrite your query and try again."
+        return result
+
+class SqlAgentAPIWrapper(BaseModel):
+    llm: BaseLanguageModel = Field(description="llm to use for sql agent")
+    sql_address: str = Field(description="sql database address for SQLDatabase uri")
+
+    db: Optional[SQLDatabase]
+    list_tables_tool: Optional[BaseTool]
+    get_schema_tool: Optional[BaseTool]
+    db_query_tool: Optional[BaseTool]
+    query_check: Optional[Any]
+    query_gen: Optional[Any]
+    workflow: Optional[StateGraph]
+    app: Optional[Any]
+
+    class Config:
+        arbitrary_types_allowed = True
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.llm = kwargs.get('llm')
+        self.sql_address = kwargs.get('sql_address')
+
+        self.db = SQLDatabase.from_uri(self.sql_address)
+        toolkit = SQLDatabaseToolkit(db=self.db, llm=self.llm)
+        tools = toolkit.get_tools()
+        self.list_tables_tool = next(tool for tool in tools if tool.name == "sql_db_list_tables")
+        self.get_schema_tool = next(tool for tool in tools if tool.name == "sql_db_schema")
+        self.db_query_tool = QueryDBTool(db=self.db)
+
+        self.query_check = self.init_query_check()
+        self.query_gen = self.init_query_gen()
+
+        # Define a new graph
+        self.workflow = StateGraph(State)
+        self.init_workflow()
+        self.app = self.workflow.compile(checkpointer=False)
+
+    def init_workflow(self):
+        self.workflow.add_node("first_tool_call", self.first_tool_call)
+        self.workflow.add_node(
+            "list_tables_tool", create_tool_node_with_fallback([self.list_tables_tool])
+        )
+
+        self.workflow.add_node("get_schema_tool", create_tool_node_with_fallback([self.get_schema_tool]))
+
+        model_get_schema = self.llm.bind_tools(
+            [self.get_schema_tool]
+        )
+        self.workflow.add_node(
+            "model_get_schema",
+            lambda state: {
+                "messages": [model_get_schema.invoke(state["messages"])],
+            },
+        )
+
+        self.workflow.add_node("query_gen", self.query_gen_node)
+        self.workflow.add_node("correct_query", self.model_check_query)
+
+        self.workflow.add_node("execute_query", create_tool_node_with_fallback([self.db_query_tool]))
+
+        self.workflow.add_edge(START, "first_tool_call")
+        self.workflow.add_edge("first_tool_call", "list_tables_tool")
+        self.workflow.add_edge("list_tables_tool", "model_get_schema")
+        self.workflow.add_edge("model_get_schema", "get_schema_tool")
+        self.workflow.add_edge("get_schema_tool", "query_gen")
+        self.workflow.add_conditional_edges(
+            "query_gen",
+            self.should_continue,
+        )
+        self.workflow.add_edge("correct_query", "execute_query")
+        self.workflow.add_edge("execute_query", "query_gen")
+
+    @staticmethod
+    def should_continue(state: State) -> Literal[END, "correct_query", "query_gen"]:
+        messages = state["messages"]
+        last_message = messages[-1]
+        # If there is a tool call, then we finish
+        if getattr(last_message, "tool_calls", None):
+            return END
+        if last_message.content.startswith("Error:"):
+            return "query_gen"
+        else:
+            return "correct_query"
+
+    def init_query_check(self):
+        query_check_system = """You are a SQL expert with a strong attention to detail.
+        Double check the SQLite query for common mistakes, including:
+        - Using NOT IN with NULL values
+        - Using UNION when UNION ALL should have been used
+        - Using BETWEEN for exclusive ranges
+        - Data type mismatch in predicates
+        - Properly quoting identifiers
+        - Using the correct number of arguments for functions
+        - Casting to the correct data type
+        - Using the proper columns for joins
+
+        If there are any of the above mistakes, rewrite the query. If there are no mistakes, just reproduce the original query.
+
+        You will call the appropriate tool to execute the query after running this check."""
+
+        query_check_prompt = ChatPromptTemplate.from_messages(
+            [("system", query_check_system), ("placeholder", "{messages}")]
+        )
+        query_check = query_check_prompt | self.llm.bind_tools(
+            [self.db_query_tool], tool_choice="required"
+        )
+        return query_check
+
+    def first_tool_call(self, state: State) -> dict[str, list[AIMessage]]:
+        return {
+            "messages": [
+                AIMessage(
+                    content="",
+                    tool_calls=[
+                        {
+                            "name": "sql_db_list_tables",
+                            "args": {},
+                            "id": "tool_abcd123",
+                        }
+                    ],
+                )
+            ]
+        }
+
+    def model_check_query(self, state: State) -> dict[str, list[AIMessage]]:
+        """
+        Use this tool to double-check if your query is correct before executing it.
+        """
+        return {"messages": [self.query_check.invoke({"messages": [state["messages"][-1]]})]}
+
+    def init_query_gen(self):
+        # Add a node for a model to generate a query based on the question and schema
+        query_gen_system = """You are a SQL expert with a strong attention to detail.Given an input question, output a syntactically correct SQL query to run, then look at the results of the query and return the answer.DO NOT call any tool besides SubmitFinalAnswer to submit the final answer.When generating the query:Output the SQL query that answers the input question without a tool call.Unless the user specifies a specific number of examples they wish to obtain, always limit your query to at most 10 results.You can order the results by a relevant column to return the most interesting examples in the database.Never query for all the columns from a specific table, only ask for the relevant columns given the question.If you get an error while executing a query, rewrite the query and try again.If you get an empty result set, you should try to rewrite the query to get a non-empty result set. NEVER make stuff up if you don't have enough information to answer the query... just say you don't have enough information.If you have enough information to answer the input question, simply invoke the appropriate tool to submit the final answer to the user.DO NOT make any DML statements (INSERT, UPDATE, DELETE, DROP etc.) to the database."""
+        query_gen_prompt = ChatPromptTemplate.from_messages(
+            [("system", query_gen_system), ("placeholder", "{messages}")]
+        )
+        query_gen = query_gen_prompt | self.llm.bind_tools(
+            [SubmitFinalAnswer]
+        )
+        return query_gen
+
+    def query_gen_node(self, state: State) -> Any:
+        message = self.query_gen.invoke(state)
+
+        # Sometimes, the LLM will hallucinate and call the wrong tool. We need to catch this and return an error message.
+        tool_messages = []
+        if message.tool_calls:
+            for tc in message.tool_calls:
+                if tc["name"] != "SubmitFinalAnswer":
+                    tool_messages.append(
+                        ToolMessage(
+                            content=f"Error: The wrong tool was called: {tc['name']}. Please fix your mistakes. Remember to only call SubmitFinalAnswer to submit the final answer. Generated queries should be outputted WITHOUT a tool call.",
+                            tool_call_id=tc["id"],
+                        )
+                    )
+        else:
+            tool_messages = []
+        return {"messages": [message] + tool_messages}
+
+    def run(self, query: str) -> str:
+        messages = self.app.invoke({"messages": [("user", query)]}, config={
+            'recursion_limit': 50
+        })
+        return messages["messages"][-1].tool_calls[0]["args"]["final_answer"]
+
+    def arun(self, query: str) -> str:
+        return self.run(query)
+
+
+class SqlAgentInput(BaseModel):
+    query: str = Field(description="用户数据查询需求（需要尽可能完整、准确）")
+
+
+class SqlAgentTool(BaseTool):
+    name = "sql_agent"
+    description = "回答与 SQL 数据库有关的问题。给定用户问题，将从数据库中获取可用的表以及对应 DDL，生成 SQL 查询语句并进行执行，最终得到执行结果。"
+    args_schema: Type[BaseModel] = SqlAgentInput
+    api_wrapper: SqlAgentAPIWrapper
+
+    def _run(
+            self,
+            query: str,
+            run_manager: Optional[CallbackManagerForToolRun] = None,
+    ) -> str:
+        """Use the tool."""
+        return self.api_wrapper.run(query)
+
+
+if __name__ == '__main__':
+    from langchain_openai import AzureChatOpenAI
+
+    llm = AzureChatOpenAI()
+    sql_agent_tool = SqlAgentTool(
+        api_wrapper=SqlAgentAPIWrapper(
+            llm=llm,
+            sql_address="sqlite:///Chinook.db",
+        )
+    )
+
+    result = sql_agent_tool.run("Which sales agent made the most in sales in 2009?")
+    print(result)

bisheng_langchain/rag/bisheng_rag_tool.py

@@ -142,6 +142,7 @@ class BishengRAGTool:
                 prompt = import_class(f'bisheng_langchain.rag.prompts.{prompt_type}')
             else:
                 prompt = None
+        self.prompt_inputs = prompt.input_variables
         self.qa_chain = create_stuff_documents_chain(llm=self.llm, prompt=prompt)
 
     def _post_init_retriever(self, retriever_type, **kwargs):
@@ -238,12 +239,12 @@ class BishengRAGTool:
             kwargs = {}
             if run_manager:
                 kwargs['config'] = RunnableConfig(callbacks=[run_manager])
-            ans = self.qa_chain.invoke(
-                {
-                    'context': docs,
-                    'question': query
-                }, **kwargs
-            )
+            tmp_input = {
+                'context': docs,
+            }
+            if 'question' in self.prompt_inputs:
+                tmp_input['question'] = query
+            ans = self.qa_chain.invoke(tmp_input, **kwargs)
         except Exception as e:
             logger.exception(f'question: {query}\nerror: {e}')
             ans = str(e)

{bisheng_langchain-0.4.0.dev1.dist-info → bisheng_langchain-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bisheng-langchain
-Version: 0.4.0.dev1
+Version: 0.4.1
 Summary: bisheng langchain modules
 Home-page: https://github.com/dataelement/bisheng
 Author: DataElem
@@ -32,7 +32,7 @@ Requires-Dist: langgraph==0.2.*
 Requires-Dist: openai==1.51.*
 Requires-Dist: langchain-openai>=0.1.25
 Requires-Dist: llama-index==0.9.48
-Requires-Dist: bisheng-ragas==1.0.0
+Requires-Dist: bisheng-ragas==1.*
 
 ## What is bisheng-langchain?
 

{bisheng_langchain-0.4.0.dev1.dist-info → bisheng_langchain-0.4.1.dist-info}/RECORD

@@ -73,11 +73,11 @@ bisheng_langchain/gpts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 bisheng_langchain/gpts/assistant.py,sha256=Xx5sNVJzB9chUHf9jYigNugWw-gyEIb6GpLOnrMWDXU,5552
 bisheng_langchain/gpts/auto_optimization.py,sha256=WNsC19rgvuDYQlSIaYThq5RqCbuobDbzCwAJW4Ksw0c,3626
 bisheng_langchain/gpts/auto_tool_selected.py,sha256=21WETf9o0YS-QEBwv3mmZRObKWszefQkXEqAA6KzoaM,1582
-bisheng_langchain/gpts/load_tools.py,sha256=U0DTN53bCXWZ26dlx9ZEPXO42Uvt72Q8mqKWcaBVQYo,8488
+bisheng_langchain/gpts/load_tools.py,sha256=CO6sdpkfUPa66Z-qZVWX34jrOlu8JBRC6j3jyYtnkjc,8755
 bisheng_langchain/gpts/message_types.py,sha256=7EJOx62j9E1U67jxWgxE_I7a8IjAvvKANknXkD2gFm0,213
 bisheng_langchain/gpts/utils.py,sha256=t3YDxaJ0OYd6EKsek7PJFRYnsezwzEFK5oVU-PRbu5g,6671
 bisheng_langchain/gpts/agent_types/__init__.py,sha256=88tFt1GfrfIqa4hCg0cMJk7rTeUmCSSdiVhR41CW4rM,381
-bisheng_langchain/gpts/agent_types/llm_functions_agent.py,sha256=A2p7-v2DLirSZ1KlkAwcR8UXkV-Tk7bkJCZ_w-_4Em4,9995
+bisheng_langchain/gpts/agent_types/llm_functions_agent.py,sha256=LRdB6QPLz6Ztqtk0pzx_nvOczlqAisAkynMoTCp1mdE,10597
 bisheng_langchain/gpts/agent_types/llm_react_agent.py,sha256=lo8Neo346aZP8tve56yiDfy6xQbtF3o_lJLIAPPgBM0,6623
 bisheng_langchain/gpts/prompts/__init__.py,sha256=pOnXvk6_PjqAoLrh68sI9o3o6znKGxoLMVFP-0XTCJo,704
 bisheng_langchain/gpts/prompts/assistant_prompt_base.py,sha256=Yp9M1XbZb5jHeBG_txcwWA84Euvl89t0g-GbJMa5Ur0,1133
@@ -92,7 +92,7 @@ bisheng_langchain/gpts/tools/api_tools/__init__.py,sha256=CkEjgIFM4GIv86V1B7SsFL
 bisheng_langchain/gpts/tools/api_tools/base.py,sha256=zPUCM_mOM9ygsc8pwejZvngEfEvGtiWTKbavfza7Eqg,3593
 bisheng_langchain/gpts/tools/api_tools/flow.py,sha256=ot2YAYgQGWgUpb2nCECAmpqHY6m0SgzwkupF9kDT3lU,2461
 bisheng_langchain/gpts/tools/api_tools/macro_data.py,sha256=FyG-qtl2ECS1CDKt6olN0eDTDM91d-UvDkMDBiVLgYQ,27429
-bisheng_langchain/gpts/tools/api_tools/openapi.py,sha256=_mvmtuWWLWYSzTsMy5K7E6zWGzpGDX3XtPrGLtTh_aU,5134
+bisheng_langchain/gpts/tools/api_tools/openapi.py,sha256=7i2Dw05u7FMKWeCKBvkNuPt9z575GaYyy-VyodUc2JM,5760
 bisheng_langchain/gpts/tools/api_tools/sina.py,sha256=4KpK7_HUUtjpdJ-K4LjPlb-occyAZcRtmmCWqJ2BotE,9708
 bisheng_langchain/gpts/tools/api_tools/tianyancha.py,sha256=abDAz-yAH1-2rKiSmZ6TgnrNUnpgAZpDY8oDiWfWapc,6684
 bisheng_langchain/gpts/tools/bing_search/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -105,6 +105,8 @@ bisheng_langchain/gpts/tools/dalle_image_generator/__init__.py,sha256=47DEQpj8HB
 bisheng_langchain/gpts/tools/dalle_image_generator/tool.py,sha256=h_mSGn2fvw4wGufrqKYC3lI1LLo9Uu_rynDM88IonMA,7631
 bisheng_langchain/gpts/tools/get_current_time/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bisheng_langchain/gpts/tools/get_current_time/tool.py,sha256=3uvk7Yu07qhZy1sBrFMhGEwyxEGMB8vubizs9x-6DG8,801
+bisheng_langchain/gpts/tools/sql_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bisheng_langchain/gpts/tools/sql_agent/tool.py,sha256=YhKfYOzy4XOMcQtW-AUe85sgWRT5k0--zW4VK6oO-Dg,11239
 bisheng_langchain/input_output/__init__.py,sha256=sW_GB7MlrHYsqY1Meb_LeimQqNsMz1gH-00Tqb2BUyM,153
 bisheng_langchain/input_output/input.py,sha256=I5YDmgbvvj1o2lO9wi8LE37wM0wP5jkhUREU32YrZMQ,1094
 bisheng_langchain/input_output/output.py,sha256=6U-az6-Cwz665C2YmcH3SYctWVjPFjmW8s70CA_qphk,11585
@@ -114,7 +116,7 @@ bisheng_langchain/rag/__init__.py,sha256=Rm_cDxOJINt0H4bOeUo3JctPxaI6xKKXZcS-R_w
 bisheng_langchain/rag/bisheng_rag_chain.py,sha256=yCgbRJ9hHOAF4lsAL2kjX-YX9J7nduIV7lsoYnuTXL4,6251
 bisheng_langchain/rag/bisheng_rag_pipeline.py,sha256=neoBK3TtuQ07_WeuJCzYlvtsDQNepUa_68NT8VCgytw,13749
 bisheng_langchain/rag/bisheng_rag_pipeline_v2.py,sha256=iOoF7mbLp9qDGPsV0fEmgph_Ba8VnECYvCPebXk8xmo,16144
-bisheng_langchain/rag/bisheng_rag_tool.py,sha256=_nTdp60_HPZmu35VmXSRoHEJyohBocUCzlAej2nHqyY,13560
+bisheng_langchain/rag/bisheng_rag_tool.py,sha256=fhvHJEJuFQ2ph50-89nTny4chy0pFxvPe-a-fdfnStI,13665
 bisheng_langchain/rag/extract_info.py,sha256=jtZ4Bchjv4tOaayC2MnkV-lLu3vDA0Hsk_S-ATni34g,1695
 bisheng_langchain/rag/run_qa_gen_web.py,sha256=-fIvHNnD3lD0iNU5m0Me1GDwRjlcsB8tE5RnPtFRG2s,1840
 bisheng_langchain/rag/run_rag_evaluate_web.py,sha256=a9vMhq-ZhEiHHr43uKUzKtjdk280uAP_UHQW_eOaQMw,2224
@@ -155,7 +157,7 @@ bisheng_langchain/vectorstores/__init__.py,sha256=zCZgDe7LyQ0iDkfcm5UJ5NxwKQSRHn
 bisheng_langchain/vectorstores/elastic_keywords_search.py,sha256=sIMbud4UfbfLTLf_pIIfcVC2lHbaTWCTTABTFP5mXkE,15334
 bisheng_langchain/vectorstores/milvus.py,sha256=jWq_lce-ihOz07D1kwj5ctPzElYexNCjJ-xSv-pK1CI,37172
 bisheng_langchain/vectorstores/retriever.py,sha256=hj4nAAl352EV_ANnU2OHJn7omCH3nBK82ydo14KqMH4,4353
-bisheng_langchain-0.4.0.dev1.dist-info/METADATA,sha256=rL3CPjFijR7QY10vULb81mpb3n80m5aWLW3Iz50b5Yo,2466
-bisheng_langchain-0.4.0.dev1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-bisheng_langchain-0.4.0.dev1.dist-info/top_level.txt,sha256=Z6pPNyCo4ihyr9iqGQbH8sJiC4dAUwA_mAyGRQB5_Fs,18
-bisheng_langchain-0.4.0.dev1.dist-info/RECORD,,
+bisheng_langchain-0.4.1.dist-info/METADATA,sha256=BFNONpW_KmqEspBNvTO9cBWIHTFkFiTdVE9--OAcCDI,2459
+bisheng_langchain-0.4.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+bisheng_langchain-0.4.1.dist-info/top_level.txt,sha256=Z6pPNyCo4ihyr9iqGQbH8sJiC4dAUwA_mAyGRQB5_Fs,18
+bisheng_langchain-0.4.1.dist-info/RECORD,,