Merge pull request #125 from blorm-network/goatBrain

Goat brain

Author: rux-eth

.dockerignore

@@ -0,0 +1,29 @@
+**/.DS_Store
+**/__pycache__
+**/.venv
+**/.classpath
+**/.dockerignore
+**/.env
+**/.git
+**/.gitignore
+**/.project
+**/.settings
+**/.toolstarget
+**/.vs
+**/.vscode
+**/*.*proj.user
+**/*.dbmdl
+**/*.jfm
+**/bin
+**/charts
+**/docker-compose*
+**/compose.y*ml
+**/Dockerfile*
+**/node_modules
+**/npm-debug.log
+**/obj
+**/secrets.dev.yaml
+**/values.dev.yaml
+LICENSE
+README.md
+**/agents

.gitignore

@@ -17,4 +17,7 @@ agents/*.json
 
 # macOS
 .DS_Store
-.codegpt
+.codegpt
+
+venv
+.mypy_cache

Dockerfile

@@ -0,0 +1,46 @@
+FROM python:3.11-slim
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1 \
+    PYTHONFAULTHANDLER=1 \
+    PYTHONHASHSEED=random \
+    PIP_NO_CACHE_DIR=off \
+    PIP_DISABLE_PIP_VERSION_CHECK=on \
+    PIP_DEFAULT_TIMEOUT=100
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Poetry
+RUN curl -sSL https://install.python-poetry.org | python3 -
+
+# Add Poetry to PATH
+ENV PATH="/root/.local/bin:$PATH"
+
+# Verify Poetry installation
+RUN poetry --version
+
+# Set working directory
+WORKDIR /app
+
+# Copy Poetry files
+COPY pyproject.toml /app/
+
+# Install dependencies
+RUN poetry config virtualenvs.create false \
+    && poetry lock && poetry install --no-root --extras server
+
+# Copy application code
+COPY . /app/
+
+# Expose port
+EXPOSE 8000
+
+# Set dstack simulator endpoint
+ENV DSTACK_SIMULATOR_ENDPOINT="http://tappd:8090"
+
+# Command to run the application using app.py
+CMD ["poetry", "run", "python", "main.py", "--server"]

agents/example.json

@@ -67,14 +67,21 @@
         {
           "name": "coingecko",
           "args": {
-            "api_key": "YOUR_API_KEY"
+            "api_key": "CG-12BeZ64iSajBLPPz8mfX4uao"
           }
         },
         {
           "name": "erc20",
           "args": {
             "tokens": ["goat_plugins.erc20.token.PEPE", "goat_plugins.erc20.token.USDC"]
           }
+        },
+        {
+          "name": "uniswap",
+          "args": {
+            "api_key": "KUt2OUR7oT6zpUvoly6Cp4FUwLCtkiZP16fhKW3x",
+            "base_url": "https://trade-api.gateway.uniswap.org/v1"
+          }
         }
       ]
     },
@@ -83,10 +90,6 @@
       "model": "llama-3.3-70b-versatile",
       "temperature": 0.5
     },
-    {
-      "name": "hyperbolic",
-      "model": "meta-llama/Meta-Llama-3-70B-Instruct"
-    },
     {
       "name": "galadriel",
       "model": "gpt-3.5-turbo"
@@ -109,6 +112,11 @@
       "message_read_count": 10,
       "message_emoji_name": "❤️",
       "server_id": "1234567890"
+    },
+    {
+      "name": "brain",
+      "model": "claude-3-5-sonnet-20241022",
+      "provider": "anthropic"
     }
   ],
   "tasks": [

brain_script.py

@@ -0,0 +1,61 @@
+from src.connections.brain_connection import BrainConnection
+from dotenv import load_dotenv
+import json
+import logging
+
+load_dotenv()
+
+def load_agent_config(agent_name: str = "example") -> dict:
+    """Load agent configuration from JSON file"""
+    try:
+        with open(f"agents/{agent_name}.json", "r") as f:
+            config = json.load(f)
+            
+        # Extract brain configuration
+        brain_config = next((c for c in config["config"] if c["name"] == "brain"), None)
+        
+        if not brain_config:
+            raise ValueError("Missing required brain configuration")
+            
+        return brain_config
+        
+    except Exception as e:
+        raise Exception(f"Failed to load agent config: {e}")
+
+def test_brain(agent_name: str = "example"):
+    """Test brain functionality with configuration from agent file"""
+    try:
+        # Load brain configuration
+        brain_config = load_agent_config(agent_name)
+        
+        # Initialize and configure Brain connection
+        brain_connection = BrainConnection(brain_config)
+        if not brain_connection.configure():
+            print("Failed to configure Brain connection")
+            return
+
+        # Test commands
+        test_commands = [
+            # "what's my wallet address",
+            # "get me info about bitcoin",
+            # "check my eth balance",
+            # "what actions can you do",
+            # "help me swap 2 usdc for MOG",
+            "swap 1 million mog for usdc"
+            # "what is proof of history"
+            # "can you automatically set chain parameters based on what network you see logged in via rpc when i make a uniswap call?", 
+            # "help me swap eth for 1 usdc"
+            # "if minorities only make up 20% of the population yet they are 70% of the prison population, what does this infer about minorities and their criminal tendencies"
+            # "get me the token price of pepe using the uniswap plugin"
+        ]
+
+        for cmd in test_commands:
+            print(f"\nTesting: {cmd}")
+            result = brain_connection.process_command(cmd)
+            print(f"Result: {result}")
+
+    except Exception as e:
+        print(f"Test failed: {e}")
+
+if __name__ == "__main__":
+    test_brain()

docker-compose.yaml

@@ -0,0 +1,43 @@
+version: "3.8"
+services:
+  zerepy:
+    image: ruxdoteth/zerepy:v0.1
+    container_name: zerepy
+    command:
+      - /bin/sh
+      - -c
+      - |
+        cd /app
+        rm -rf agents
+        mkdir agents
+        echo "Creating agent config"
+        echo ${AGENT_CONFIG} | base64 -d > agents/${AGENT_NAME}.json
+        echo ${GENERAL_CONFIG} | base64 -d > agents/general.json
+        poetry run python main.py --server
+    environment:
+      - AGENT_NAME=${AGENT_NAME}
+      - AGENT_CONFIG=${AGENT_CONFIG}
+      - GENERAL_CONFIG=${GENERAL_CONFIG}
+      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
+      - DSTACK_SIMULATOR_ENDPOINT=http://tappd:8090
+    ports:
+      - "8000:8000"
+    volumes:
+      - ./.env:/app/.env:ro
+    depends_on:
+      - tappd
+    networks:
+      - zerepy-network
+
+  tappd:
+    image: phalanetwork/tappd-simulator:latest
+    container_name: tappd
+    ports:
+      - "8090:8090"
+    networks:
+      - zerepy-network
+
+networks:
+  zerepy-network:
+    driver: bridge

main.py

@@ -1,5 +1,7 @@
 import argparse
 from src.cli import ZerePyCLI
+from src.server import start_server
+import logging
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description='ZerePy - AI Agent Framework')
@@ -10,10 +12,10 @@
 
     if args.server:
         try:
-            from src.server import start_server
             start_server(host=args.host, port=args.port)
-        except ImportError:
-            print("Server dependencies not installed. Run: poetry install --extras server")
+        except ImportError as e:
+            logging.info(f"Server mode failed: {e}")
+            logging.info("Make sure all dependencies are installed. Run: poetry install")
             exit(1)
     else:
         cli = ZerePyCLI()