Add SQL agent to AgentQnA (#1370)

Signed-off-by: minmin-intel <minmin.hou@intel.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: chen, suyue <suyue.chen@intel.com>

AgentQnA/tools/tools.py

@@ -8,13 +8,30 @@ from tools.pycragapi import CRAG
 
 
 def search_knowledge_base(query: str) -> str:
-    """Search the knowledge base for a specific query."""
-    # use worker agent (DocGrader) to search the knowledge base
+    """Search a knowledge base about music and singers for a given query.
+
+    Returns text related to the query.
+    """
     url = os.environ.get("WORKER_AGENT_URL")
     print(url)
     proxies = {"http": ""}
     payload = {
-        "query": query,
+        "messages": query,
+    }
+    response = requests.post(url, json=payload, proxies=proxies)
+    return response.json()["text"]
+
+
+def search_sql_database(query: str) -> str:
+    """Search a SQL database on artists and their music with a natural language query.
+
+    Returns text related to the query.
+    """
+    url = os.environ.get("SQL_AGENT_URL")
+    print(url)
+    proxies = {"http": ""}
+    payload = {
+        "messages": query,
     }
     response = requests.post(url, json=payload, proxies=proxies)
     return response.json()["text"]