single_iteration_anthropic.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 1] - Introducción\n",
    "# Single iteration example\n",
    "\"\"\"\n",
    "Una vez que estés satisfecho con la primera versión de tu funcionalidad, puedes construir \n",
    "un planificador alrededor de esto para ejecutar regularmente o escuchar eventos.\n",
    "\"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 2] - Setup del cliente Anthropic\n",
    "# DIFERENCIA: Aquí usamos anthropic en lugar de openai\n",
    "import os\n",
    "from dotenv import load_dotenv\n",
    "from anthropic import Anthropic  # Nueva importación\n",
    "load_dotenv()\n",
    "\n",
    "# DIFERENCIA: Configuración específica para Anthropic\n",
    "anthropic_client = Anthropic(\n",
    "    api_key=os.getenv(\"ANTHROPIC_API_KEY\"),  # Usa ANTHROPIC_API_KEY en lugar de OPENAI_API_KEY\n",
    ")\n",
    "\n",
    "# Si obtienes un error aquí, asegúrate de que:\n",
    "# 1. Hayas instalado el paquete anthropic: pip install anthropic\n",
    "# 2. Tengas un archivo .env con ANTHROPIC_API_KEY=\"tu_clave_api_aqui\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{\n",
      "  \"marginSummary\": {\n",
      "    \"accountValue\": \"327149.602201\",\n",
      "    \"totalNtlPos\": \"44260.82962\",\n",
      "    \"totalRawUsd\": \"352872.449681\",\n",
      "    \"totalMarginUsed\": \"2229.320572\"\n",
      "  },\n",
      "  \"crossMarginSummary\": {\n",
      "    \"accountValue\": \"327149.602201\",\n",
      "    \"totalNtlPos\": \"44260.82962\",\n",
      "    \"totalRawUsd\": \"352872.449681\",\n",
      "    \"totalMarginUsed\": \"2229.320572\"\n",
      "  },\n",
      "  \"crossMaintenanceMarginUsed\": \"442.608293\",\n",
      "  \"withdrawable\": \"324920.281629\",\n",
      "  \"assetPositions\": [\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"ATOM\",\n",
      "        \"szi\": \"491.45\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"10.9725\",\n",
      "        \"positionValue\": \"3064.6822\",\n",
      "        \"unrealizedPnl\": \"-2327.752925\",\n",
      "        \"returnOnEquity\": \"-8.63340169\",\n",
      "        \"liquidationPx\": null,\n",
      "        \"marginUsed\": \"153.23411\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"-1615.831717\",\n",
      "          \"sinceOpen\": \"-1615.831717\",\n",
      "          \"sinceChange\": \"87.67426\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"BTC\",\n",
      "        \"szi\": \"-0.15792\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"31538.2\",\n",
      "        \"positionValue\": \"14630.18256\",\n",
      "        \"unrealizedPnl\": \"-9649.670016\",\n",
      "        \"returnOnEquity\": \"-38.7497067\",\n",
      "        \"liquidationPx\": \"2140972.98477735\",\n",
      "        \"marginUsed\": \"731.509128\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"-7674.268201\",\n",
      "          \"sinceOpen\": \"-7674.268201\",\n",
      "          \"sinceChange\": \"-7876.893803\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"ETH\",\n",
      "        \"szi\": \"-2.8225\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"2008.46\",\n",
      "        \"positionValue\": \"9526.502\",\n",
      "        \"unrealizedPnl\": \"-3857.62365\",\n",
      "        \"returnOnEquity\": \"-13.60983042\",\n",
      "        \"liquidationPx\": \"117980.07907743\",\n",
      "        \"marginUsed\": \"476.3251\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"-8974.821362\",\n",
      "          \"sinceOpen\": \"-8974.821362\",\n",
      "          \"sinceChange\": \"-9135.179456\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"BNB\",\n",
      "        \"szi\": \"-14.872\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"346.33\",\n",
      "        \"positionValue\": \"10540.82744\",\n",
      "        \"unrealizedPnl\": \"-5390.202207\",\n",
      "        \"returnOnEquity\": \"-20.93028308\",\n",
      "        \"liquidationPx\": \"22459.19167806\",\n",
      "        \"marginUsed\": \"527.041372\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"-2537.526657\",\n",
      "          \"sinceOpen\": \"-2537.526657\",\n",
      "          \"sinceChange\": \"-2553.471438\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"GMT\",\n",
      "        \"szi\": \"232.0\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 2\n",
      "        },\n",
      "        \"entryPx\": \"0.323488\",\n",
      "        \"positionValue\": \"36.17576\",\n",
      "        \"unrealizedPnl\": \"-38.873456\",\n",
      "        \"returnOnEquity\": \"-1.03594569\",\n",
      "        \"liquidationPx\": null,\n",
      "        \"marginUsed\": \"18.08788\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"184.804378\",\n",
      "          \"sinceOpen\": \"184.804378\",\n",
      "          \"sinceChange\": \"2.087772\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"DYDX\",\n",
      "        \"szi\": \"-200.7\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"3.0287\",\n",
      "        \"positionValue\": \"294.32655\",\n",
      "        \"unrealizedPnl\": \"313.53354\",\n",
      "        \"returnOnEquity\": \"10.31597715\",\n",
      "        \"liquidationPx\": \"1613.18683481\",\n",
      "        \"marginUsed\": \"14.716327\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"-1120.419938\",\n",
      "          \"sinceOpen\": \"-46.359904\",\n",
      "          \"sinceChange\": \"-46.359904\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"APE\",\n",
      "        \"szi\": \"2689.3\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"3.82611\",\n",
      "        \"positionValue\": \"3206.18346\",\n",
      "        \"unrealizedPnl\": \"-7083.374163\",\n",
      "        \"returnOnEquity\": \"-13.76808299\",\n",
      "        \"liquidationPx\": null,\n",
      "        \"marginUsed\": \"160.309173\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"456.386527\",\n",
      "          \"sinceOpen\": \"456.386527\",\n",
      "          \"sinceChange\": \"456.386527\"\n",
      "        }\n",
      "      }\n",
      "    },\n",
      "    {\n",
      "      \"type\": \"oneWay\",\n",
      "      \"position\": {\n",
      "        \"coin\": \"OP\",\n",
      "        \"szi\": \"1646.9\",\n",
      "        \"leverage\": {\n",
      "          \"type\": \"cross\",\n",
      "          \"value\": 20\n",
      "        },\n",
      "        \"entryPx\": \"2.0938\",\n",
      "        \"positionValue\": \"2961.94965\",\n",
      "        \"unrealizedPnl\": \"-486.346038\",\n",
      "        \"returnOnEquity\": \"-2.8207908\",\n",
      "        \"liquidationPx\": null,\n",
      "        \"marginUsed\": \"148.097482\",\n",
      "        \"maxLeverage\": 50,\n",
      "        \"cumFunding\": {\n",
      "          \"allTime\": \"961.167796\",\n",
      "          \"sinceOpen\": \"961.167796\",\n",
      "          \"sinceChange\": \"803.3661\"\n",
      "        }\n",
      "      }\n",
      "    }\n",
      "  ],\n",
      "  \"time\": 1735617939001\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "# [Celda 3] - Setup Hyperliquid SDK\n",
    "# Esta parte no cambia respecto a las versiones anteriores\n",
    "from hyperliquid.info import Info\n",
    "from hyperliquid.utils import constants\n",
    "import json\n",
    "\n",
    "info = Info(constants.TESTNET_API_URL, skip_ws=True)\n",
    "\n",
    "# Esta es una dirección de ejemplo\n",
    "user_state = info.user_state(\"0xcd5051944f780a621ee62e39e493c489668acf4d\") \n",
    "print(json.dumps(user_state, indent=2))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 4] - Definición de identidad del usuario\n",
    "# Esta parte no cambia\n",
    "identity_map = {\n",
    "    \"risk_level\": \"Moderate risk, willing to risk some money for the right investments but not chasing every new opportunity.\",\n",
    "    \"token_preferences\": \"Likes ETH more than BTC, doesnt like SOL\",\n",
    "    \"mission_statement\": \"Accumulate as much ETH as possible given the available funds.\",\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Available balance: 324920.281629 USD\n",
      "Current ATOM position: size 491.45 ATOM, value of 3064.6822 usd, leverage 20, and unrealizedPnl -2327.752925 usd. The max leverage for this position is 50.\n",
      "Current BTC position: size -0.15792 BTC, value of 14630.18256 usd, leverage 20, and unrealizedPnl -9649.670016 usd. The max leverage for this position is 50.\n",
      "Current ETH position: size -2.8225 ETH, value of 9526.502 usd, leverage 20, and unrealizedPnl -3857.62365 usd. The max leverage for this position is 50.\n",
      "Current BNB position: size -14.872 BNB, value of 10540.82744 usd, leverage 20, and unrealizedPnl -5390.202207 usd. The max leverage for this position is 50.\n",
      "Current GMT position: size 232.0 GMT, value of 36.17576 usd, leverage 2, and unrealizedPnl -38.873456 usd. The max leverage for this position is 50.\n",
      "Current DYDX position: size -200.7 DYDX, value of 294.32655 usd, leverage 20, and unrealizedPnl 313.53354 usd. The max leverage for this position is 50.\n",
      "Current APE position: size 2689.3 APE, value of 3206.18346 usd, leverage 20, and unrealizedPnl -7083.374163 usd. The max leverage for this position is 50.\n",
      "Current OP position: size 1646.9 OP, value of 2961.94965 usd, leverage 20, and unrealizedPnl -486.346038 usd. The max leverage for this position is 50.\n",
      "\n"
     ]
    }
   ],
   "source": [
    "# [Celda 5] - Recopilación de datos on-chain\n",
    "# Esta parte no cambia\n",
    "available_balance = user_state.get('withdrawable')\n",
    "print(f\"Available balance: {available_balance} USD\")\n",
    "\n",
    "positions = user_state.get('assetPositions')\n",
    "positions_for_llm = ''\n",
    "if positions:\n",
    "    for position in positions:\n",
    "        position = position.get('position')\n",
    "        position_for_llm = f\"Current {position.get('coin')} position: size {position.get('szi')} {position.get('coin')}, value of {position.get('positionValue')} usd, leverage {position.get('leverage').get('value')}, and unrealizedPnl {position.get('unrealizedPnl')} usd. The max leverage for this position is {position.get('maxLeverage')}.\\n\"\n",
    "        positions_for_llm += position_for_llm\n",
    "print(positions_for_llm)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 6] - Información adicional\n",
    "# Esta parte no cambia\n",
    "recent_headlines = (\"*ROBINHOOD: CRYPTO TRADING VOLUMES OVER $30B, UP 600% YOY\\n\"\n",
    "                   \"*FED'S MUSALEM: TIME MAY BE APPROACHING TO SLOW OR PAUSE RATE CUTS\\n\")\n",
    "\n",
    "chat_summary = \"Bob is happy with the Ethereum roadmap and has been hearing more people talk about it.\\n\"\n",
    "\n",
    "chat_and_data_summary = chat_summary + recent_headlines"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 7] - Definición de prompts\n",
    "# El contenido es similar, pero el formato es específico para Anthropic\n",
    "system_prompt = (\n",
    "    \"# Instructions:\\n\"\n",
    "    \"Act as a helpful cryptocurrency assistant helping users manage their trading portfolio.\\n\"\n",
    "    \"They understand it is inherently risky to trade cryptocurrency, and they want to make sure they are making informed decisions.\\n\"\n",
    "    # ... [resto del system_prompt igual que antes] ...\n",
    ")\n",
    "\n",
    "user_message = (\n",
    "    \"# Instructions:\\n\"\n",
    "    \"Here are some details about me, can you help me make decisions about my trading portfolio?\\n\"\n",
    "    f\"# Risk Level\\n{identity_map.get('risk_level')}\\n\"\n",
    "    f\"# Available Balance\\n{available_balance}\\n\"\n",
    "    f\"# Open Positions\\n{positions_for_llm}\\n\"\n",
    "    f\"# Recent Information\\n{chat_and_data_summary}\\n\"\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 8] - Definición del formato de respuesta\n",
    "# Esta parte no cambia\n",
    "from pydantic import BaseModel, Field\n",
    "from typing import Literal\n",
    "\n",
    "class Position(BaseModel):\n",
    "    market: str = Field(..., description=\"The asset to trade\")\n",
    "    direction: Literal[\"long\", \"short\"] = Field(..., description=\"The direction to trade\")\n",
    "    size: float = Field(..., description=\"The size of the trade denominated in USD. It should be greater than 10.\")\n",
    "    reasoning: list[str] = Field(default_factory=list, description=\"The reasoning for the decision\")\n",
    "    leverage: int | None = Field(None, description=\"Optional leverage multiplier\")\n",
    "\n",
    "class PositionReasoning(BaseModel):\n",
    "    positions_to_maintain: list[Position] = Field(default_factory=list)\n",
    "    positions_to_modify: list[Position] = Field(default_factory=list)\n",
    "    positions_to_open: list[Position] = Field(default_factory=list)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "metadata": {},
   "outputs": [],
   "source": [
    "# [Celda 9] - Llamada a la API\n",
    "# Modificamos el mensaje del usuario para solicitar específicamente un formato JSON\n",
    "user_message_with_format = (\n",
    "    user_message + \"\\n\\n\"\n",
    "    \"Please respond ONLY with a JSON object that follows this exact structure, with no additional text before or after:\\n\"\n",
    "    \"{\\n\"\n",
    "    '  \"positions_to_maintain\": [],\\n'\n",
    "    '  \"positions_to_modify\": [],\\n'\n",
    "    '  \"positions_to_open\": []\\n'\n",
    "    \"}\\n\"\n",
    "    \"Each position should have: market (string), direction ('long' or 'short'), size (float > 10), reasoning (array of strings), and optional leverage (integer).\"\n",
    ")\n",
    "\n",
    "messages = [\n",
    "    {\n",
    "        \"role\": \"user\",\n",
    "        \"content\": user_message_with_format\n",
    "    }\n",
    "]\n",
    "\n",
    "# DIFERENCIA: Formato específico de Anthropic para la llamada\n",
    "\n",
    "# Claude 3.5 Sonnet claude-3-5-sonnet-20241022\n",
    "# Claude 3.5 Haiku claude-3-5-haiku-20241022\n",
    "# Claude 3 Opus claude-3-opus-20240229\n",
    "# Claude 3 Sonnet claude-3-sonnet-20240229 OK\n",
    "# Claude 3 Haiku claude-3-haiku-20240307\n",
    "\n",
    "completion = anthropic_client.messages.create(\n",
    "    model=\"claude-3-opus-20240229\",\n",
    "    max_tokens=4096,\n",
    "    messages=messages,\n",
    "    system=system_prompt,\n",
    "    temperature=0.7\n",
    ")\n",
    "\n",
    "# DIFERENCIA: Procesamiento de la respuesta de Anthropic\n",
    "try:\n",
    "    # Extraemos el texto del TextBlock\n",
    "    response_text = completion.content[0].text\n",
    "    \n",
    "    # Intentamos encontrar y extraer solo la parte JSON\n",
    "    import re\n",
    "    json_match = re.search(r'(\\{[\\s\\S]*\\})', response_text)\n",
    "    if json_match:\n",
    "        json_str = json_match.group(1)\n",
    "        result = PositionReasoning.model_validate_json(json_str)\n",
    "    else:\n",
    "        raise ValueError(\"No JSON found in response\")\n",
    "        \n",
    "except Exception as e:\n",
    "    print(f\"Error processing response: {e}\")\n",
    "    print(\"\\nRaw response:\")\n",
    "    print(completion.content)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "Positions to maintain:\n",
      "- ETH: short -2.8225 (leverage: None)\n",
      "  Reasoning: Despite positive Ethereum roadmap news, maintaining short position until clear trend reversal\n",
      "\n",
      "Positions to modify:\n",
      "- BTC: short -0.08 (leverage: 10)\n",
      "  Reasoning: Reduce exposure due to high unrealized losses, Lower leverage to decrease risk, Fed comments suggest potential monetary policy shift\n",
      "- APE: long 1000.0 (leverage: 10)\n",
      "  Reasoning: Current position showing significant losses, Reduce position size and leverage to manage risk\n",
      "\n",
      "Positions to open:\n",
      "- ETH: long 15.0 (leverage: 5)\n",
      "  Reasoning: Positive Ethereum roadmap developments, Increased market interest, Lower leverage for risk management\n"
     ]
    }
   ],
   "source": [
    "# [Celda 10] - Visualización de resultados\n",
    "# Nueva celda para mejor visualización\n",
    "print(\"\\nPositions to maintain:\")\n",
    "for pos in result.positions_to_maintain:\n",
    "    print(f\"- {pos.market}: {pos.direction} {pos.size} (leverage: {pos.leverage})\")\n",
    "    print(f\"  Reasoning: {', '.join(pos.reasoning)}\")\n",
    "\n",
    "print(\"\\nPositions to modify:\")\n",
    "for pos in result.positions_to_modify:\n",
    "    print(f\"- {pos.market}: {pos.direction} {pos.size} (leverage: {pos.leverage})\")\n",
    "    print(f\"  Reasoning: {', '.join(pos.reasoning)}\")\n",
    "\n",
    "print(\"\\nPositions to open:\")\n",
    "for pos in result.positions_to_open:\n",
    "    print(f\"- {pos.market}: {pos.direction} {pos.size} (leverage: {pos.leverage})\")\n",
    "    print(f\"  Reasoning: {', '.join(pos.reasoning)}\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}