add tools scripts for agents

Signed-off-by: cheehook <[email protected]>

Author: cheehook

app-backend/templates/tools/supervisor_agent.yaml

@@ -0,0 +1,20 @@
+ # Copyright (C) 2024 Intel Corporation
+ # SPDX-License-Identifier: Apache-2.0
+ 
+ search_knowledge_base:
+   description: Search a knowledge base for a given query. Returns text related to the query.
+   callable_api: tools.py:search_knowledge_base
+   args_schema:
+     query:
+       type: str
+       description: query
+   return_output: retrieved_data
+ 
+ search_sql_database:
+   description: Search a SQL database with a natural language query. Returns text related to the query.
+   callable_api: tools.py:search_sql_database
+   args_schema:
+     query:
+       type: str
+       description: natural language query
+   return_output: retrieved_data

app-backend/templates/tools/tools.py

@@ -0,0 +1,35 @@
+# Copyright (C) 2024 Intel Corporation
+ # SPDX-License-Identifier: Apache-2.0
+ 
+import os
+import requests
+
+
+def search_knowledge_base(query: str) -> str:
+    """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 = {
+        "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"]

app-backend/templates/tools/worker_agent_tools.py

@@ -0,0 +1,40 @@
+# Copyright (C) 2024 Intel Corporation
+ # SPDX-License-Identifier: Apache-2.0
+ 
+import os
+import requests
+
+def search_knowledge_base(query: str) -> str:
+    """Search the knowledge base for a specific query."""
+    url = os.environ.get("RETRIEVAL_TOOL_URL")
+    print(url)
+    proxies = {"http": ""}
+    payload = {
+        "text": query,
+    }
+    response = requests.post(url, json=payload, proxies=proxies)
+    print(response)
+    if "documents" in response.json():
+        docs = response.json()["documents"]
+        context = ""
+        for i, doc in enumerate(docs):
+            if i == 0:
+                context = doc
+            else:
+                context += "\n" + doc
+        # print(context)
+        return context
+    elif "text" in response.json():
+        return response.json()["text"]
+    elif "reranked_docs" in response.json():
+        docs = response.json()["reranked_docs"]
+        context = ""
+        for i, doc in enumerate(docs):
+            if i == 0:
+                context = doc["text"]
+            else:
+                context += "\n" + doc["text"]
+        # print(context)
+        return context
+    else:
+        return "Error parsing response from the knowledge base."

app-backend/templates/tools/worker_agent_tools.yaml

@@ -0,0 +1,11 @@
+# Copyright (C) 2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+search_knowledge_base:
+  description: Search knowledge base for a given query. Returns text related to the query.
+  callable_api: worker_agent_tools.py:search_knowledge_base
+  args_schema:
+    query:
+      type: str
+      description: query
+  return_output: retrieved_data