debugging za history

src/agent.py

@@ -62,61 +62,55 @@ def chat_with_turbot(message: str, history: List[List[str]]) -> str:
     """
     Main function to chat with TurBot.
     """
-    # Convert history to the format expected by our LLM
-    messages = []
+    # Prepare the system message first
+    system_message = """Ti si TurBot, digitalni asistent za turističku agenciju. 
+    Odgovori na srpskom jeziku i budi koristan i prijateljski nastrojen. 
+    Ako ne znaš odgovor na pitanje, preporuči korisniku da pita o putovanjima, letovima, hotelima ili destinacijama."""
+    
+    # Convert messages to the format expected by InferenceClient
+    messages_for_client = []
+    
+    # Add system message first (only once)
+    messages_for_client.append({"role": "system", "content": system_message})
+    
+    # Add conversation history
     for human, ai in history:
         if human:
-            messages.append(HumanMessage(content=human))
+            messages_for_client.append({"role": "user", "content": human})
         if ai:
-            messages.append(AIMessage(content=ai))
+            messages_for_client.append({"role": "assistant", "content": ai})
     
     # Add the current message
-    messages.append(HumanMessage(content=message))
+    messages_for_client.append({"role": "user", "content": message})
     
     # Check if we should search for travel information
     context = ""
     if should_search_travel_info(message):
         context = search_travel_packages(message)
-    
-    # Prepare the system message
-    if context:
-        system_message = f"""Ti si TurBot, digitalni asistent za turističku agenciju. 
-        Koristi sledeće informacije o putovanjima da odgovoriš na pitanje korisnika:
-        
-        {context}
-        
-        Odgovori na srpskom jeziku i budi koristan i prijateljski nastrojen."""
-    else:
-        system_message = """Ti si TurBot, digitalni asistent za turističku agenciju. 
-        Odgovori na srpskom jeziku i budi koristan i prijateljski nastrojen. 
-        Ako ne znaš odgovor na pitanje, preporuči korisniku da pita o putovanjima, letovima, hotelima ili destinacijama."""
-    
-    # Add system message to the beginning
-    messages.insert(0, HumanMessage(content=system_message))
+        if context:
+            # Add context as additional information
+            messages_for_client.append({"role": "user", "content": f"Kontekst o putovanjima: {context}"})
     
     try:
-        # Convert messages to the format expected by InferenceClient
-        messages_for_client = []
-        for msg in messages:
-            if isinstance(msg, HumanMessage):
-                messages_for_client.append({"role": "user", "content": msg.content})
-            elif isinstance(msg, AIMessage):
-                messages_for_client.append({"role": "assistant", "content": msg.content})
+        # Debug: Print messages being sent to model
+        print(f"\n🔍 DEBUG: Sending {len(messages_for_client)} messages to model:")
+        for i, msg in enumerate(messages_for_client):
+            print(f"  {i+1}. {msg['role']}: {msg['content'][:100]}...")
         
         # Generate response
         response = ""
-        for message in client.chat_completion(
+        for message_chunk in client.chat_completion(
             messages_for_client,
             stream=True,
             max_tokens=512,
             temperature=0.7,
             top_p=0.95,
         ):
-            if hasattr(message, 'choices') and message.choices:
-                token = message.choices[0].delta.content
+            if hasattr(message_chunk, 'choices') and message_chunk.choices:
+                token = message_chunk.choices[0].delta.content
                 if token:
                     response += token
         
-        return response
+        return response if response else "Izvinjavam se, nisam dobio odgovor od modela. Pokušajte ponovo."
     except Exception as e:
         return f"Izvinjavam se, došlo je do greške: {str(e)}" 

test_agent.py

@@ -1,10 +1,48 @@
 #!/usr/bin/env python3
 """
-Simple test script for TurBot agent.
+Simple test script for TurBot agent with memory debugging.
 """
 
 from src.agent import chat_with_turbot
 
+def test_memory():
+    """Test the TurBot agent's memory with name test."""
+    
+    print("🧠 Testing TurBot Memory - Name Test")
+    print("=" * 60)
+    
+    test_questions = [
+        "Zdravo! Moje ime je Marko.",
+        "Kako se zovem?",
+        "Da li se sećaš mog imena?",
+        "Koje je moje ime?"
+    ]
+    
+    history = []
+    
+    for i, question in enumerate(test_questions):
+        print(f"\n👤 Korisnik: {question}")
+        print("-" * 40)
+        
+        try:
+            response = chat_with_turbot(question, history)
+            print(f"🤖 TurBot: {response}")
+            
+            # Add to history for next iteration
+            history.append([question, response])
+            
+            # Debug: Show current history
+            print(f"\n📝 History length: {len(history)}")
+            print("📝 Current history:")
+            for j, (h, a) in enumerate(history):
+                print(f"  {j+1}. User: {h[:50]}...")
+                print(f"     Bot: {a[:50]}...")
+            
+        except Exception as e:
+            print(f"❌ Greška: {str(e)}")
+        
+        print("-" * 40)
+
 def test_turbot():
     """Test the TurBot agent with some sample questions."""
     
@@ -37,4 +75,12 @@ def test_turbot():
         print("-" * 40)
 
 if __name__ == "__main__":
+    # Run memory test first
+    test_memory()
+    
+    print("\n" + "="*60)
+    print("Now testing regular questions...")
+    print("="*60)
+    
+    # Then run regular test
     test_turbot()