mezoAgent 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

{mezoAgent-0.4.0.dist-info → mezoAgent-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.4.0
+Version: 0.4.2
 Summary: A Python package for Mezo Agent transactions with LangChain tools
 Author: Dreadwulf, Duck, Digi
 Requires-Dist: python-dotenv

{mezoAgent-0.4.0.dist-info → mezoAgent-0.4.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
-mezo_agent/__init__.py,sha256=-UKUAyREgtXorqjaWebWM20KQsUxcP-IwzB0LQV5fKY,305
+mezo_agent/__init__.py,sha256=PRFZfcq8B68pQI2toci1ACM_SPcu7suy9D3OD8IvZeI,450
 mezo_agent/characters.py,sha256=wub7y9BrAAdAKWwcNAdgrvhRi2zHMaZNzTr2y3qtbsw,515
-mezo_agent/chat.py,sha256=p0oNWvgmpHoNzSHjjDtQ_H8xVfqzvcdjv_Nk56YfFYg,1096
+mezo_agent/chat.py,sha256=ZtNurUDV7UzJjbFyU96DNGy37hO3gl0osn19mRu7ER8,859
 mezo_agent/config.py,sha256=1PWeaHdQPKqKPraE-tM81CwRKeBiCbz5C_8iZTEiCgo,2148
 mezo_agent/parsing.py,sha256=rolPVE8r_RS6BqKe25r1nyZQ47rvTPG2Hv09Kxxfv4c,2821
 mezo_agent/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
 mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
 mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.4.0.dist-info/METADATA,sha256=Fe9uL9lkeAWFe9S0wHXv4aYJVkhXBWK-8OdytVmdRB8,273
-mezoAgent-0.4.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.4.0.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.4.0.dist-info/RECORD,,
+mezoAgent-0.4.2.dist-info/METADATA,sha256=YEbcKgTvvW13zDdNJKRz47odDmCOX73-Bl_KTa3Qn_k,273
+mezoAgent-0.4.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.4.2.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.4.2.dist-info/RECORD,,

mezo_agent/__init__.py

@@ -1,6 +1,12 @@
 from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
 from .swap_musd_btc import mezo_agent_swap_musd_btc
-from .chat import mezo_chat, mezo_agent_chat_tool
+from .chat import mezo_character_chat  # Corrected function name
 from .characters import get_character_prompt
 
-__all__ = ["mezo_chat", "mezo_agent_chat_tool", "get_character_prompt"]
+__all__ = [
+    "mezo_agent_transaction_btc",
+    "mezo_agent_musd_transaction",
+    "mezo_agent_swap_musd_btc",
+    "mezo_character_chat",  # Corrected function name
+    "get_character_prompt"
+]

mezo_agent/chat.py

@@ -1,12 +1,13 @@
-from mezo_agent.config import Config  # Use existing config file
 from mezo_agent.characters import get_character_prompt  # Import character selection
 from langchain_openai import ChatOpenAI
-from langchain.tools import Tool
+from langchain.tools import tool
+from .config import OPENAI_API_KEY
 
 # Initialize OpenAI Chat Model
-llm = ChatOpenAI(temperature=0, openai_api_key=Config.OPENAI_API_KEY)
+llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
 
-def mezo_chat(prompt: str, character: str = "crypto_native") -> str:
+@tool
+def mezo_character_chat(prompt: str, character: str = "DigAIJoe") -> str:
     """
     Generates a chatbot response based on the user's input and the selected AI character.
     
@@ -19,9 +20,3 @@ def mezo_chat(prompt: str, character: str = "crypto_native") -> str:
     response = llm.invoke(query)
     return response.content.strip()
 
-# Define the chat tool for LangChain
-mezo_agent_chat_tool = Tool(
-    name="Mezo Chat Tool",
-    func=mezo_chat,
-    description="Chat with an AI character that responds based on a selected personality."
-)