updated to dev

app.py

@@ -28,6 +28,7 @@ load_dotenv()
 
 Defaults.drivers_config = AnthropicDriversConfig()
 
+
 base_url = "https://cloud.griptape.ai"
 
 headers_api = {
@@ -81,17 +82,8 @@ def bot(history):
 def create_prompt_task(session_id: str, message: str) -> PromptTask:
     return PromptTask(
         f"""
-            Re-structure the values from the user's questions: '{message}' and the input value from the conversation memory to fit the following format. Leave out attributes that aren't important to the user: 
-                years experience: <x>
-                location: <x>
-                role: <x>
-                skills: <x>
-                expected salary: <x>
-                availability: <x>
-                past companies: <x>
-                past projects: <x>
-                show reel details: <x>
-            """,
+            Re-structure the values to form a query from the user's questions: '{message}' and the input value from the conversation memory. Leave out attributes that aren't important to the user: 
+        """,
     )
 
 
@@ -106,29 +98,29 @@ def build_talk_agent(session_id: str, message: str) -> Agent:
                 value="You are responsible for structuring a user's questions into a specific format for a query."
             ),
             Rule(
-                value="You ask the user follow-up questions to fill in missing information for the format you are trying to fit."
-            ),
-            Rule(
-                value="If the user has no preference for a specific attribute, then you can remove it from the query."
+                value="""You ask the user follow-up questions to fill in missing information for:
+                years experience, 
+                location, 
+                role, 
+                skills, 
+                expected salary,
+                availability,
+                past companies,
+                past projects,
+                show reel details
+                """
             ),
             Rule(
-                value="Only return the current query structure and any questions to fill in missing information."
+                value="Return the current query structure and any questions to fill in missing information."
             ),
         ],
     )
-    file_manager_tool = FileManagerTool(
-        name="FileManager",
-        file_manager_driver=LocalFileManagerDriver(),
-        off_prompt=False,
-    )
-
     return Agent(
         conversation_memory=ConversationMemory(
-            driver=GriptapeCloudConversationMemoryDriver(
-                thread_id=str(threads[session_id]),
+            conversation_memory_driver=GriptapeCloudConversationMemoryDriver(
+                thread_id=threads[session_id],
             )
         ),
-        tools=[file_manager_tool],
         tasks=[create_prompt_task(session_id, message)],
         rulesets=[ruleset],
     )
@@ -144,7 +136,7 @@ def build_agent(session_id: str, message: str) -> Agent:
         name="Local Gradio Agent",
         rules=[
             Rule(
-                value="You are responsible for structuring a user's questions into a specific format for a query and then querying."
+                value="You are responsible for structuring a user's questions into a query and then querying."
             ),
             Rule(
                 value="Only return the result of the query, do not provide additional commentary."
@@ -154,7 +146,7 @@ def build_agent(session_id: str, message: str) -> Agent:
                 value="Do not perform the query unless the user has said 'Done' with formulating."
             ),
             Rule(
-                value="Only perform the query with the proper query structure as one string argument."
+                value="Only perform the query as one string argument."
             ),
             Rule(
                 value="If you reformulate the query, then you must ask the user if they are 'Done' again."
@@ -165,16 +157,13 @@ def build_agent(session_id: str, message: str) -> Agent:
         ],
     )
 
-    print("Base URL", os.environ.get("BASE_URL", "https://cloud.griptape.ai"))
-
     query_client = StructureRunTool(
         name="QueryResumeSearcher",
         description="Use it to search for a candidate with the query.",
         driver=GriptapeCloudStructureRunDriver(
-            # base_url=os.environ.get("BASE_URL","https://cloud.griptape.ai"),
             structure_id=os.getenv("GT_STRUCTURE_ID"),
             api_key=os.getenv("GT_CLOUD_API_KEY"),
-            structure_run_wait_time_interval=5,
+            structure_run_wait_time_interval=3,
             structure_run_max_wait_time_attempts=30,
         ),
     )
@@ -186,10 +175,9 @@ def build_agent(session_id: str, message: str) -> Agent:
             structure_factory_fn=lambda: build_talk_agent(session_id, message),
         ),
     )
-
     return Agent(
         conversation_memory=ConversationMemory(
-            driver=GriptapeCloudConversationMemoryDriver(
+            conversation_memory_driver=GriptapeCloudConversationMemoryDriver(
                 thread_id=threads[session_id],
             )
         ),

pyproject.toml

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