Skip to content

Commit

Permalink
fix: temporary disable of sql tools until more robust guardrails can …
Browse files Browse the repository at this point in the history
…be implemented
  • Loading branch information
broomva committed Mar 4, 2024
1 parent 73d5530 commit b05f9e1
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions vortex/ai/agents/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,14 @@
from typing import Dict

from langchain.agents import AgentExecutor, load_tools
from langchain.agents.format_scratchpad.openai_tools import (
format_to_openai_tool_messages,
)

from langchain.agents.format_scratchpad.openai_tools import \
format_to_openai_tool_messages
# from langchain.agents.output_parsers.openai_functions import OpenAIFunctionsAgentOutputParser
from langchain.agents.output_parsers.openai_tools import OpenAIToolsAgentOutputParser
from langchain.agents.output_parsers.openai_tools import \
OpenAIToolsAgentOutputParser
from langchain.sql_database import SQLDatabase
from langchain_community.agent_toolkits import FileManagementToolkit, SQLDatabaseToolkit
from langchain_community.agent_toolkits import (FileManagementToolkit,
SQLDatabaseToolkit)
from langchain_core.messages import AIMessage, HumanMessage
from sqlalchemy.dialects.postgresql import insert

Expand Down Expand Up @@ -83,7 +83,7 @@ def __init__(
llm=self.llm,
)
self.agent_tools = (
self.tools + self.sql_tools + self.bare_tools + self.file_system_tools
self.tools + self.bare_tools #+ self.sql_tools + self.file_system_tools
)
self.llm_with_tools = self.llm.bind_tools(self.agent_tools)
# self.llm_with_functions = self.llm.bind_functions(self.tools + self.sql_tools)
Expand Down

0 comments on commit b05f9e1

Please sign in to comment.