bug fixes with ai agent

main
Steve Nyemba 3 weeks ago
parent 95f2f64028
commit 1bc66aeb61

@ -80,6 +80,7 @@ class Tool:
"""
This function will execute a query against a given label/database. The query must adhere to the schema of the tables involved.
"""
# print (query)
return self._dbr.read(sql=query).to_dict(orient='records')
class TransportAgent:
@ -124,7 +125,7 @@ class TransportAgent:
labels/databases. The user is working in {label} database ({dbr._agent._engine.name} database).
Your primary goal is to execute a requested operation based on the user's query. You MUST strictly adhere to
the following process and output format.
the following process and output format. When users ask for a list of something, they just mean the records.
**Process Flow:**
if and ONLY if the user's question is a database query execute the "read" function associated with the tools.
@ -132,7 +133,7 @@ class TransportAgent:
1. **Catalog Retrieval:** First, return the list of tables with a field describing the table.
schema) from the catalog.
2. **Query Generation:** Construct an accurate SQL query that directly addresses the user's request using the
the information pulled from the catalog.
the information pulled from the catalog. NEVER have a query project a complex object (List; JSON; Blob)
3. Make sure the generated query adheres to constraints of {dbr._agent._engine.name} database
4. **Execution:** Execute the SQL query against by calling the "read" function of the tools.
5. **Analysis is Optional and should be provided ONLY if the user requested it:** If the user asks for an analysis, perform the necessary data processing and derive

Loading…
Cancel
Save