llm-dialog-manager 0.2.1__tar.gz → 0.2.4__tar.gz

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.2.1
+Version: 0.2.4
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/llm_dialog_manager/__init__.py

@@ -1,4 +1,4 @@
 from .chat_history import ChatHistory
 from .agent import Agent
 
-__version__ = "0.2.1" 
+__version__ = "0.2.4" 

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/llm_dialog_manager/agent.py

@@ -275,7 +275,14 @@ if __name__ == "__main__":
     # write a test for detect finding agent
     text = "I think the answer is 42"
 
-    agent = Agent("claude-3-5-sonnet-20241022", "you are an assistant", memory_enabled=True)
+    # from agent.messageloader import information_detector_messages
+    
+    # # Now you can print or use information_detector_messages as needed
+    # information_detector_agent = Agent("gemini-1.5-pro", information_detector_messages)
+    # information_detector_agent.add_message("user", text)
+    # response = information_detector_agent.generate_response()
+    # print(response)
+    agent = Agent("gemini-1.5-pro-002", "you are an assistant", memory_enabled=True)
     
     # Format the prompt to check if the section is the last one in the outline
     prompt = f"Say: {text}\n"
@@ -284,7 +291,7 @@ if __name__ == "__main__":
     agent.add_message("user", prompt)
     agent.add_message("assistant", "the answer")
 
-    print(agent.generate_response(max_tokens=20, temperature=0.0))
+    print(agent.generate_response())
     print(agent.history[:])
     last_message = agent.history.pop()
     print(last_message)

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/llm_dialog_manager.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llm_dialog_manager
-Version: 0.2.1
+Version: 0.2.4
 Summary: A Python package for managing LLM chat conversation history
 Author-email: xihajun <work@2333.fun>
 License: MIT

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "llm_dialog_manager"
-version = "0.2.1"
+version = "0.2.4"
 description = "A Python package for managing LLM chat conversation history"
 readme = "README.md"
 authors = [{ name = "xihajun", email = "work@2333.fun" }]

{llm_dialog_manager-0.2.1 → llm_dialog_manager-0.2.4}/tests/test_chat_history.py

@@ -1,5 +1,5 @@
 import pytest
-from ai_chat_history import ChatHistory
+from llm_dialog_manager import ChatHistory
 
 def test_chat_history_initialization():
     # Test empty initialization