Initial commit.
Basic docker deployment with Local LLM integration and simple game state.
This commit is contained in:
53
test_llm_exchange.py
Normal file
53
test_llm_exchange.py
Normal file
@@ -0,0 +1,53 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script for basic message exchange with LLM.
|
||||
"""
|
||||
|
||||
from config import Config
|
||||
from llm_client import LLMClient
|
||||
from conversation import ConversationManager
|
||||
|
||||
|
||||
def test_message_exchange():
|
||||
"""Test basic message exchange with LLM."""
|
||||
print("Testing message exchange with LLM...")
|
||||
print("-" * 40)
|
||||
|
||||
# Create components
|
||||
config = Config()
|
||||
llm_client = LLMClient(config)
|
||||
conversation_manager = ConversationManager()
|
||||
|
||||
# Test connection first
|
||||
try:
|
||||
success = llm_client.test_connection()
|
||||
if not success:
|
||||
print("✗ Failed to connect to LM Studio")
|
||||
return
|
||||
else:
|
||||
print("✓ Connected to LM Studio")
|
||||
except Exception as e:
|
||||
print(f"✗ Error testing connection: {e}")
|
||||
return
|
||||
|
||||
# Add a test message to conversation
|
||||
test_message = "Hello, are you there?"
|
||||
conversation_manager.add_user_message(test_message)
|
||||
print(f"Sending message: {test_message}")
|
||||
|
||||
# Get response from LLM
|
||||
try:
|
||||
response = llm_client.get_response(conversation_manager.get_history())
|
||||
print(f"Received response: {response}")
|
||||
|
||||
# Add response to conversation
|
||||
conversation_manager.add_assistant_message(response)
|
||||
|
||||
print("\n✓ Message exchange successful!")
|
||||
|
||||
except Exception as e:
|
||||
print(f"✗ Error in message exchange: {e}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_message_exchange()
|
||||
Reference in New Issue
Block a user