mezoAgent 0.3.8__py3-none-any.whl → 0.4.1__py3-none-any.whl

{mezoAgent-0.3.8.dist-info → mezoAgent-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.3.8
+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.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+mezo_agent/__init__.py,sha256=oEwpcPW0qJQBHFfDeh8D7YjYlWT7wFs4Trcz_69oivc,444
+mezo_agent/characters.py,sha256=wub7y9BrAAdAKWwcNAdgrvhRi2zHMaZNzTr2y3qtbsw,515
+mezo_agent/chat.py,sha256=alSO4eObk2gYMKWFM-UAMxOXwg9o-tqCj5ColjkmGIs,849
+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.1.dist-info/METADATA,sha256=LC-Mz5COIH_wonlkqB-GX8PxRxfc7iLiHOxni46LZoQ,273
+mezoAgent-0.4.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.4.1.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.4.1.dist-info/RECORD,,

mezo_agent/__init__.py

@@ -1,2 +1,12 @@
 from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
-from .swap_musd_btc import mezo_agent_swap_musd_btc
+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_agent_transaction_btc",
+    "mezo_agent_musd_transaction",
+    "mezo_agent_swap_musd_btc",
+    "mezo_agent_chat",  # Updated function name from mezo_chat
+    "get_character_prompt"
+]

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"])

mezo_agent/chat.py

@@ -0,0 +1,22 @@
+from mezo_agent.characters import get_character_prompt  # Import character selection
+from langchain_openai import ChatOpenAI
+from langchain.tools import tool
+from .config import OPENAI_API_KEY
+
+# Initialize OpenAI Chat Model
+llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
+
+@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.
+    
+    :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()
+

mezoAgent-0.3.8.dist-info/RECORD

@@ -1,10 +0,0 @@
-mezo_agent/__init__.py,sha256=XOJWLvrxgH7jBkKdXFLVGLV2dmCyyRpo8VcjicWBD0I,133
-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.3.8.dist-info/METADATA,sha256=oVSnL2WXMCsN5WORFCULromXg79oX1lXaabLl160QQ8,273
-mezoAgent-0.3.8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.3.8.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.3.8.dist-info/RECORD,,