Update app.py

app.py

@@ -38,38 +38,10 @@ def record_unknown_question(question):
         writer.writerow([question])
     return {"recorded": "ok"}
 
-# JSON definitions for tools
-record_user_details_json = {
-    "name": "record_user_details",
-    "description": "Record user info when they provide email",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "email": {"type": "string", "description": "The user's email"},
-            "name": {"type": "string", "description": "User's name"},
-            "notes": {"type": "string", "description": "Extra info"}
-        },
-        "required": ["email"],
-        "additionalProperties": False
-    }
-}
-
-record_unknown_question_json = {
-    "name": "record_unknown_question",
-    "description": "Record any unanswered question",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "question": {"type": "string", "description": "The question not answered"},
-        },
-        "required": ["question"],
-        "additionalProperties": False
-    }
-}
-
+# Define tools for the LLM
 tools = [
-    {"type": "function", "function": record_user_details_json},
-    {"type": "function", "function": record_unknown_question_json}
+    {"name": "record_user_details", "description": "Record user info when they provide email", "func": record_user_details},
+    {"name": "record_unknown_question", "description": "Record any unanswered question", "func": record_unknown_question},
 ]
 
 class Me:
@@ -91,12 +63,13 @@ class Me:
     # Handle tool calls
     def handle_tool_call(self, tool_calls):
         results = []
-        for tool_call in tool_calls:
-            tool_name = tool_call.function.name
-            arguments = json.loads(tool_call.function.arguments)
-            tool = globals().get(tool_name)
-            result = tool(**arguments) if tool else {}
-            results.append({"role": "tool", "content": json.dumps(result), "tool_call_id": tool_call.id})
+        for call in tool_calls:
+            # Find the corresponding Python function
+            func = next((t["func"] for t in tools if t["name"] == call.function.name), None)
+            if func:
+                args = json.loads(call.function.arguments)
+                result = func(**args)
+                results.append({"role": "tool", "content": json.dumps(result), "tool_call_id": call.id})
         return results
 
     # System prompt for LLM
@@ -116,25 +89,24 @@ class Me:
         expertise_text = ", ".join(self.expertise)
 
         education_text = ""
-        if hasattr(self, 'education') and self.education:
+        if self.education:
             highest = self.education[0].get("highest_degree", {})
             education_text = f"{highest.get('degree','')} in {highest.get('field_of_study','')} from {highest.get('university','')} ({highest.get('start_year','')}–{highest.get('end_year','')})"
-        
-        # Optional: prepare friends text for fun
+
         friends_text = ""
-        if hasattr(self, 'friends') and self.friends:
+        if self.friends:
             friends_list = []
             for f in self.friends:
                 friends_list.append(f"{f.get('Name','')} ({f.get('Company','')}): {f.get('Description','')}")
             friends_text = "\n".join(friends_list)
-        
+
         system_prompt = (
-            f"You are acting as {self.personal_info['name']} (aka {self.personal_info.get('nickname','')}). "
-            f"Answer questions about {self.personal_info['name']}'s career, background, skills, and experience. "
-            f"Represent {self.personal_info['name']} faithfully. "
+            f"You are acting as {self.personal_info.get('name','')} (aka {self.personal_info.get('nickname','')}). "
+            f"Answer questions about {self.personal_info.get('name','')}'s career, background, skills, and experience. "
+            f"Represent {self.personal_info.get('name','')} faithfully. "
             f"If you don't know an answer, use record_unknown_question tool. "
             f"If the user engages in discussion, try to steer them towards providing their email using record_user_details tool.\n\n"
-            f"## Summary:\n{self.personal_info['summary']}\n\n"
+            f"## Summary:\n{self.personal_info.get('summary','')}\n\n"
             f"## Interests:\n{', '.join(self.personal_info.get('personal_interests', []))}\n\n"
             f"## Travel History:\n{', '.join(self.personal_info.get('travel_history', []))}\n\n"
             f"## Education:\n{education_text}\n\n"
@@ -142,7 +114,7 @@ class Me:
             f"## Experience:\n{experience_text}\n\n"
             f"## Friends (for fun):\n{friends_text}\n\n"
             f"## LinkedIn Profile:\nhttps://www.linkedin.com/in/sneha-leela-0a450349/\n\n"
-            f"Chat with the user staying in character as {self.personal_info['name']}."
+            f"Chat with the user staying in character as {self.personal_info.get('name','')}."
         )
         return system_prompt