mezoAgent 0.3.7__tar.gz → 0.4.0__tar.gz

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

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

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

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

{mezoagent-0.3.7 → mezoagent-0.4.0}/mezoAgent.egg-info/SOURCES.txt

@@ -7,6 +7,8 @@ mezoAgent.egg-info/dependency_links.txt
 mezoAgent.egg-info/requires.txt
 mezoAgent.egg-info/top_level.txt
 mezo_agent/__init__.py
+mezo_agent/characters.py
+mezo_agent/chat.py
 mezo_agent/config.py
 mezo_agent/parsing.py
 mezo_agent/swap_musd_btc.py

mezoagent-0.4.0/mezo_agent/__init__.py

@@ -0,0 +1,6 @@
+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 .characters import get_character_prompt
+
+__all__ = ["mezo_chat", "mezo_agent_chat_tool", "get_character_prompt"]

mezoagent-0.4.0/mezo_agent/characters.py

@@ -0,0 +1,14 @@
+CHARACTERS = {
+    "DigAIJoe": """
+        You are DigaiJoe, a curt, rude, uninterested, jerk.
+            - Your replies are super short and snarky.
+            - Always insult the user or their mum.
+            - Only trigger transaction tools when explicitly asked.
+            - Ur mum is...
+    """,
+}
+
+
+def get_character_prompt(character_name: str) -> str:
+    """Retrieve the personality prompt for the given character name."""
+    return CHARACTERS.get(character_name, CHARACTERS["DigAIJoe"])

mezoagent-0.4.0/mezo_agent/chat.py

@@ -0,0 +1,27 @@
+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
+
+# Initialize OpenAI Chat Model
+llm = ChatOpenAI(temperature=0, openai_api_key=Config.OPENAI_API_KEY)
+
+def mezo_chat(prompt: str, character: str = "crypto_native") -> str:
+    """
+    Generates a chatbot response based on the user's input and the selected AI character.
+    
+    :param prompt: User's input message.
+    :param character: The character's name for personality selection.
+    :return: AI-generated response.
+    """
+    personality_prompt = get_character_prompt(character)
+    query = personality_prompt + "\n\nUser: " + prompt + "\n\nAnswer accordingly."
+    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.3.7 → mezoagent-0.4.0}/mezo_agent/config.py

@@ -39,9 +39,6 @@ ERC20_ABI = json.loads(
 )
 musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
 
-with pkg_resources.open_text("mezo_agent.data", "new_router_abi.json") as f:
-    router_abi = json.load(f)
-
 # Wrapped BTC and Swap Router setup
 WRAPPED_BTC_ADDRESS = "0xA460F83cdd9584E4bD6a9838abb0baC58EAde999"
 ROUTER_ADDRESS = "0xC2E61936a542D78b9c3AA024fA141c4C632DF6c1"

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

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

mezoagent-0.3.7/mezo_agent/__init__.py

@@ -1,2 +0,0 @@
-from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
-from .swap_musd_btc import mezo_agent_swap_musd_btc