llm-dialog-manager 0.1.2480__py3-none-any.whl → 0.2.7__py3-none-any.whl

llm_dialog_manager/__init__.py

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

llm_dialog_manager/agent.py

@@ -117,7 +117,6 @@ def completion(model: str, messages: List[Dict[str, str]], max_tokens: int = 100
                     api_key=api_key,
                     base_url="https://generativelanguage.googleapis.com/v1beta/"
                 )
-                print(api_key)
                 # Remove any system message from the beginning if present
                 if messages and messages[0]["role"] == "system":
                     system_msg = messages.pop(0)
@@ -131,7 +130,6 @@ def completion(model: str, messages: List[Dict[str, str]], max_tokens: int = 100
                     # max_tokens=max_tokens,
                     temperature=temperature
                 )
-                print(len(response.choices))
                 
                 return response.choices[0].message.content
 
@@ -249,13 +247,6 @@ if __name__ == "__main__":
     # write a test for detect finding agent
     text = "I think the answer is 42"
 
-    # 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("claude-3-5-sonnet-20241022", "you are an assistant", memory_enabled=True)
     
     # Format the prompt to check if the section is the last one in the outline
@@ -267,6 +258,6 @@ if __name__ == "__main__":
 
     print(agent.generate_response(max_tokens=20, temperature=0.0))
     print(agent.history[:])
-    a = agent.history.pop()
-    print(a)
+    last_message = agent.history.pop()
+    print(last_message)
     print(agent.history[:])

{llm_dialog_manager-0.1.2480.dist-info → llm_dialog_manager-0.2.7.dist-info}/METADATA

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

llm_dialog_manager-0.2.7.dist-info/RECORD

@@ -0,0 +1,9 @@
+llm_dialog_manager/__init__.py,sha256=bQb_jouDWIWf_n5x2kBA7uc43xkG2IMLeYcV1IevNek,86
+llm_dialog_manager/agent.py,sha256=FF8vs-RyJGKWit5X82eLHyel1sThf76sLEa_1GVLeA0,9868
+llm_dialog_manager/chat_history.py,sha256=xKA-oQCv8jv_g8EhXrG9h1S8Icbj2FfqPIhbty5vra4,6033
+llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
+llm_dialog_manager-0.2.7.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
+llm_dialog_manager-0.2.7.dist-info/METADATA,sha256=kmdHmOHId-OSQxMBeDbtdAn7eWz-VGJfK_cBGs5BjqU,5193
+llm_dialog_manager-0.2.7.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+llm_dialog_manager-0.2.7.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
+llm_dialog_manager-0.2.7.dist-info/RECORD,,

llm_dialog_manager-0.1.2480.dist-info/RECORD

@@ -1,9 +0,0 @@
-llm_dialog_manager/__init__.py,sha256=AfSsuE8ty-luFdLkdgQ98oQfbOYqYM8SO6p4tMDcDZ4,89
-llm_dialog_manager/agent.py,sha256=dExW-pfPpRLedi-iJsn5S_iImMmWGI1ixztVqWCmMHE,10303
-llm_dialog_manager/chat_history.py,sha256=xKA-oQCv8jv_g8EhXrG9h1S8Icbj2FfqPIhbty5vra4,6033
-llm_dialog_manager/key_manager.py,sha256=shvxmn4zUtQx_p-x1EFyOmnk-WlhigbpKtxTKve-zXk,4421
-llm_dialog_manager-0.1.2480.dist-info/LICENSE,sha256=vWGbYgGuWpWrXL8-xi6pNcX5UzD6pWoIAZmcetyfbus,1064
-llm_dialog_manager-0.1.2480.dist-info/METADATA,sha256=1MvkDLGrjfAsNe8HTAK0_BNNTMLO_UPZEmi27pGujkY,5196
-llm_dialog_manager-0.1.2480.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-llm_dialog_manager-0.1.2480.dist-info/top_level.txt,sha256=u2EQEXW0NGAt0AAHT7jx1odXZ4rZfjcgbmJhvKFuMkI,19
-llm_dialog_manager-0.1.2480.dist-info/RECORD,,