Upload main.py

main.py

@@ -43,22 +43,59 @@ async def startup():
     reload_knowledge()
 
 def get_stem(word):
-    # Einfaches Stemming für bessere Treffer
     w = word.lower().strip()
     for end in ["ern", "en", "er", "es", "st", "te", "e", "s", "t"]:
         if w.endswith(end) and len(w) > len(end)+2: return w[:-len(end)]
     return w
 
+# --- HELPER: VAPI REQUEST PARSER ---
+# Holt sicher die ID und die Argumente aus dem Vapi-Salat
+def parse_vapi_request(data):
+    tool_call_id = "unknown"
+    args = {}
+    
+    try:
+        msg = data.get("message", {})
+        
+        # Variante A: Dein Log-Format ("toolCallList")
+        if "toolCallList" in msg:
+            call = msg["toolCallList"][0]
+            tool_call_id = call["id"]
+            if "function" in call and "arguments" in call["function"]:
+                args = call["function"]["arguments"]
+        
+        # Variante B: Standard Vapi ("toolCalls")
+        elif "toolCalls" in msg:
+            call = msg["toolCalls"][0]
+            tool_call_id = call["id"]
+            if "function" in call and "arguments" in call["function"]:
+                args = call["function"]["arguments"]
+                
+        # Argumente sind oft ein String, müssen zu JSON werden
+        if isinstance(args, str):
+            args = json.loads(args)
+            
+    except Exception as e:
+        print(f"⚠️ Parsing Info: {e}")
+        
+    return tool_call_id, args
+
 # ==========================================
-# TOOL 1: VERFÜGBARKEIT (Bleibt wie es ist)
+# TOOL 1: VERFÜGBARKEIT (Strict Vapi Format)
 # ==========================================
 @app.post("/check_availability")
 async def check_availability(request: Request):
-    print("🚦 TOOL CALL: checkAvailability")
+    data = await request.json()
+    print(f"🚦 TOOL REQUEST: {json.dumps(data)[:100]}...") # Debug Log
+
+    # 1. ID extrahieren (WICHTIG!)
+    tool_call_id, _ = parse_vapi_request(data)
+    
     today = datetime.now().strftime("%Y-%m-%d")
     status = "available"
-    message = "Normaler Betrieb."
+    instruction = "Normal arbeiten"
     
+    # 2. Datenbank Check
     try:
         if db:
             rules = db.collection(COLLECTION_RULES).where("active", "==", True).stream()
@@ -67,92 +104,82 @@ async def check_availability(request: Request):
                 if rd.get('start_date') <= today <= rd.get('end_date'):
                     print(f"🛑 REGEL AKTIV: {rd.get('name')}")
                     status = "unavailable"
-                    message = rd.get('instruction_text')
+                    instruction = rd.get('instruction_text')
                     break
     except Exception as e:
         print(f"❌ ERROR CHECK: {e}")
 
+    print(f"👉 RESULTAT: {status} | {instruction}")
+    
+    # 3. Antwort im strikten Vapi-Format
     return {
-        "result": json.dumps({
-            "status": status,
-            "instruction": message
-        })
+        "results": [
+            {
+                "toolCallId": tool_call_id,
+                "result": {
+                    "status": status,
+                    "instruction": instruction
+                }
+            }
+        ]
     }
 
 # ==========================================
-# TOOL 2: SUCHE (Jetzt mit Udo & Capaneo Priorität)
+# TOOL 2: SUCHE (Jetzt auch mit ID Rückgabe!)
 # ==========================================
 @app.post("/search")
 async def search(request: Request):
-    try:
-        data = await request.json()
-        query = ""
-        
-        # Parsing
-        if "search_query" in data: query = data["search_query"]
-        elif "message" in data and "toolCalls" in data["message"]:
-            args = data["message"]["toolCalls"][0]["function"]["arguments"]
-            if isinstance(args, str): args = json.loads(args)
-            query = args.get("search_query") or args.get("query")
-
-        print(f"🔎 FRAGE: '{query}'")
-        if not query: return {"result": "Akustik-Fehler."}
-
-        # --- UPDATE: STOP WÖRTER BEREINIGT ---
-        # Wir entfernen "capaneo" und "udo" hier! Sie sind wichtig!
-        # Nur Füllwörter bleiben gesperrt.
-        STOP_WORDS = ["hallo", "guten", "tag", "moin", "bitte", "danke", "frage", "kannst", "sagen"]
-        
+    data = await request.json()
+    
+    # 1. ID und Query extrahieren
+    tool_call_id, args = parse_vapi_request(data)
+    query = args.get("search_query") or args.get("query") or data.get("search_query")
+    
+    print(f"🔎 FRAGE (ID: {tool_call_id}): '{query}'")
+    
+    answer_text = "Dazu habe ich keine Informationen."
+    
+    if query:
+        # --- SUCHE LOGIK (Mit Udo/Capaneo Boost) ---
+        STOP_WORDS = ["hallo", "guten", "tag", "moin", "bitte", "danke", "frage"]
         q_words = [get_stem(w) for w in query.lower().split() if len(w)>2]
         relevant_words = [w for w in q_words if w not in STOP_WORDS]
-
-        if not relevant_words:
-             return {"result": "Dazu habe ich keine Informationen."}
         
-        best_doc = None
-        best_score = 0
-
-        for doc in KNOWLEDGE_CACHE:
-            score = 0
-            # Wir holen Titel (Frage) und Inhalt (Antwort)
-            title = doc.get("question", "").lower()
-            content = doc.get("answer", "").lower()
-            keywords = [k.lower() for k in doc.get("keywords", [])]
-            
-            for word in relevant_words:
-                word_stem = get_stem(word)
-
-                # 1. TITEL TREFFER (Das Wichtigste!)
-                # Wenn der Kunde "Capaneo" sagt und im Titel steht "Wer ist Capaneo" -> BINGO!
-                if word in title: 
-                    score += 50 
-                
-                # 2. KEYWORD TREFFER
-                # Keywords sind auch stark
-                for k in keywords:
-                    if get_stem(k) == word_stem:
-                        score += 30
+        if relevant_words:
+            best_doc = None
+            best_score = 0
+            for doc in KNOWLEDGE_CACHE:
+                score = 0
+                title = doc.get("question", "").lower()
+                content = doc.get("answer", "").lower()
+                keywords = [k.lower() for k in doc.get("keywords", [])]
                 
-                # 3. TEXT TREFFER (Nur Beiwerk)
-                # Verhindert, dass jedes Dokument mit "Capaneo" im Footer gewinnt
-                if word in content:
-                    score += 5
-
-            if score > best_score:
-                best_score = score
-                best_doc = doc
-
-        # Score Schwelle auf 20 (damit "Hallo" keine Treffer bringt, aber "Udo" schon)
-        if best_doc and best_score >= 20:
-            print(f"🏆 TREFFER ({best_score} Pkt): {best_doc.get('question')}")
-            return {"result": best_doc.get("answer")}
-        else:
-            print(f"⚠️ Zu wenig Relevanz (Max Score: {best_score})")
-            return {"result": "Dazu habe ich leider keine spezifischen Informationen in meiner Datenbank."}
-
-    except Exception as e:
-        print(f"❌ SEARCH ERROR: {e}")
-        return {"result": "Fehler."}
+                for word in relevant_words:
+                    if word in title: score += 50 
+                    for k in keywords:
+                        if get_stem(k) == get_stem(word): score += 30
+                    if word in content: score += 5
+
+                if score > best_score:
+                    best_score = score
+                    best_doc = doc
+
+            if best_doc and best_score >= 20:
+                print(f"🏆 TREFFER ({best_score}): {best_doc.get('question')}")
+                answer_text = best_doc.get("answer")
+            else:
+                 print(f"⚠️ Zu wenig Relevanz (Max: {best_score})")
+
+    # 3. Antwort im strikten Vapi-Format
+    # Bei 'search' erwartet das LLM meist einfach den Text im result
+    return {
+        "results": [
+            {
+                "toolCallId": tool_call_id,
+                "result": answer_text 
+            }
+        ]
+    }
 
 # ==========================================
 # 3. MÜLLSCHLUCKER