updating to just directly call the structure

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 from typing import Any
 from dotenv import load_dotenv
 import requests
-from griptape.structures import Agent
-from griptape.tasks import PromptTask
+from griptape.structures import Agent, Structure, Workflow
+from griptape.tasks import PromptTask, StructureRunTask
 from griptape.drivers import (
     LocalConversationMemoryDriver,
     GriptapeCloudStructureRunDriver,
@@ -134,7 +134,7 @@ def build_talk_agent(session_id: str, message: str) -> Agent:
 
 # Creates an agent for each run
 # The agent uses local memory, which it differentiates between by session_hash.
-def build_agent(session_id: str, message: str, kbs:str) -> Agent:
+def build_agent(session_id: str, message: str, kbs: str) -> Agent:
 
     create_thread_id(session_id)
 
@@ -151,15 +151,13 @@ def build_agent(session_id: str, message: str, kbs:str) -> Agent:
             Rule(
                 value="Do not perform the query unless the user has confirmed they are done with formulating."
             ),
-            Rule(
-                value="Only perform the query as one string argument."
-            ),
+            Rule(value="Only perform the query as one string argument."),
             Rule(
                 value="If the user says they want to start over, then you must delete the conversation memory file."
             ),
             Rule(
                 value="Do not ever search conversation memory for a formulated query instead of querying. Query every time."
-            )
+            ),
         ],
     )
 
@@ -172,13 +170,16 @@ def build_agent(session_id: str, message: str, kbs:str) -> Agent:
             structure_run_wait_time_interval=3,
             structure_run_max_wait_time_attempts=30,
         ),
+        # structure_run_driver = LocalStructureRunDriver(
+        #     create_structure=create_structure
+        # )
     )
 
     talk_client = StructureRunTool(
         name="FormulateQueryFromUser",
         description="Used to formulate a query from the user's input.",
-        driver=LocalStructureRunDriver(
-            structure_factory_fn=lambda: build_talk_agent(session_id, message),
+        structure_run_driver=LocalStructureRunDriver(
+            create_structure=lambda: build_talk_agent(session_id, message),
         ),
     )
     return Agent(
@@ -199,11 +200,67 @@ def send_message(message: str, history, knowledge_bases, request: gr.Request) ->
     response = agent.run(message)
     return response.output.value
 
+
+def send_message_call(message: str, history, knowledge_bases) -> Any:
+
+    structure_id = os.getenv("GT_STRUCTURE_ID")
+    api_key = os.getenv("GT_CLOUD_API_KEY")
+    structure_url = f"https://cloud.griptape.ai/api/structures/{structure_id}/runs"
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+    payload = {"args": [message, *knowledge_bases]}
+    response = requests.post(structure_url, headers=headers, json=payload)
+    response.raise_for_status()
+    if response.status_code == 201:
+        data = response.json()
+        structure_run_id = data["structure_run_id"]
+        output = poll_structure(structure_run_id, headers)
+        return output["output_task_output"]["value"]
+    else:
+        return "Assistant Call Failed"
+
+
+def poll_for_events(offset: int, structure_run_id: str, headers: dict):
+    url = f"https://cloud.griptape.ai/api/structure-runs/{structure_run_id}/events"
+    response = requests.get(
+        url=url, headers=headers, params={"offset": offset, "limit": 100}
+    )
+    response.raise_for_status()
+
+    return response
+
+
+def poll_structure(structure_run_id: str, headers: dict):
+    response = poll_for_events(0, structure_run_id, headers)
+    events = response.json()["events"]
+    offset = response.json()["next_offset"]
+    not_finished = True
+    output = ""
+    while not_finished:
+        time.sleep(0.5)
+        for event in events:
+            if event["type"] == "FinishStructureRunEvent":
+                not_finished = False
+                output = dict(event["payload"])
+                break
+        response = response = poll_for_events(offset, structure_run_id, headers)
+        response.raise_for_status()
+        events = response.json()["events"]
+        offset = response.json()["next_offset"]
+    return output
+
+
 with gr.Blocks() as demo:
-    knowledge_bases = gr.CheckboxGroup(label="Select Knowledge Bases", choices=["skills","demographics","linked_in","showreels"])
-    chatbot = gr.ChatInterface(fn=send_message, chatbot=gr.Chatbot(height=300),additional_inputs=knowledge_bases)
-demo.launch(auth=(os.environ.get("GRADIO_USERNAME"), os.environ.get("GRADIO_PASSWORD")))
-#demo.launch()
+    knowledge_bases = gr.CheckboxGroup(
+        label="Select Knowledge Bases",
+        choices=["skills", "demographics", "linked_in", "showreels"],
+    )
+    chatbot = gr.ChatInterface(
+        fn=send_message_call,
+        chatbot=gr.Chatbot(height=300),
+        additional_inputs=knowledge_bases,
+    )
+# demo.launch(auth=(os.environ.get("GRADIO_USERNAME"), os.environ.get("GRADIO_PASSWORD")))
+demo.launch()
 
 # Set it back to empty when a session is done
 # Is there a better way?

pyproject.toml

@@ -13,6 +13,7 @@ python = "^3.11"
 python-dotenv = "^1.0.0"
 gradio = "^4.37.1"
 griptape = {git="https://github.com/griptape-ai/griptape.git", rev = "dev", extras=["drivers-embedding-voyageai","drivers-prompt-anthropic"]}
+#griptape = "^0.34"
 argparse = "^1.4.0"
 azure-identity = "^1.17.1"