Spaces:
Sleeping
Sleeping
Update src/agent_graph/tool_travel_sqlagent.py
Browse files
src/agent_graph/tool_travel_sqlagent.py
CHANGED
|
@@ -21,7 +21,7 @@ class TravelSQLAgentTool:
|
|
| 21 |
generate a final answer for the user.
|
| 22 |
|
| 23 |
Attributes:
|
| 24 |
-
sql_agent_llm (
|
| 25 |
system_role (str): A system prompt template that guides the language model in answering user questions based on SQL query results.
|
| 26 |
db (SQLDatabase): An instance of the SQL database used to execute queries.
|
| 27 |
chain (RunnablePassthrough): A chain of operations that creates SQL queries, executes them, and generates a response.
|
|
|
|
| 21 |
generate a final answer for the user.
|
| 22 |
|
| 23 |
Attributes:
|
| 24 |
+
sql_agent_llm (ChatGroq): An instance of a ChatGroq language model used to generate and process SQL queries.
|
| 25 |
system_role (str): A system prompt template that guides the language model in answering user questions based on SQL query results.
|
| 26 |
db (SQLDatabase): An instance of the SQL database used to execute queries.
|
| 27 |
chain (RunnablePassthrough): A chain of operations that creates SQL queries, executes them, and generates a response.
|