mezoAgent 0.4.0__tar.gz → 0.4.1__tar.gz

{mezoagent-0.4.0 → mezoagent-0.4.1}/PKG-INFO

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

{mezoagent-0.4.0 → mezoagent-0.4.1}/mezoAgent.egg-info/PKG-INFO

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

{mezoagent-0.4.0 → mezoagent-0.4.1}/mezo_agent/__init__.py

@@ -3,4 +3,10 @@ from .swap_musd_btc import mezo_agent_swap_musd_btc
 from .chat import mezo_chat, mezo_agent_chat_tool
 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_agent_chat",  # Updated function name from mezo_chat
+    "get_character_prompt"
+]

{mezoagent-0.4.0 → mezoagent-0.4.1}/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_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."
-)

{mezoagent-0.4.0 → mezoagent-0.4.1}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="mezoAgent",
-    version="0.4.0",
+    version="0.4.1",
     packages=find_packages(),
     include_package_data=True,
     package_data={