Spaces:
Sleeping
Sleeping
Priyanshi Saxena
commited on
Commit
Β·
f104fee
1
Parent(s):
20eee66
feat: added api tools
Browse files- .env.example +1 -1
- .gitignore +29 -0
- Dockerfile +18 -0
- README.md +66 -0
- app.py +123 -284
- minimal_test.py +36 -0
- requirements.txt +14 -7
- run.py +47 -0
- src/agent/__init__.py +0 -0
- src/agent/query_planner.py +89 -0
- src/agent/research_agent.py +138 -0
- src/api/__init__.py +0 -0
- src/api/airaa_integration.py +93 -0
- src/config.py +7 -0
- src/tools/__init__.py +0 -0
- src/tools/base_tool.py +64 -0
- src/tools/coingecko_tool.py +116 -0
- src/tools/defillama_tool.py +129 -0
- src/tools/etherscan_tool.py +138 -0
- src/utils/__init__.py +0 -0
- src/utils/config.py +1 -0
- src/utils/exceptions.py +16 -0
- src/utils/logger.py +17 -0
- src/visualizations.py +41 -217
- test_app.py +108 -0
.env.example
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Google Gemini API Key (Required)
|
| 2 |
-
GEMINI_API_KEY=
|
| 3 |
|
| 4 |
# CoinGecko API Key (Optional - for higher rate limits)
|
| 5 |
COINGECKO_API_KEY=your_coingecko_api_key_here
|
|
|
|
| 1 |
# Google Gemini API Key (Required)
|
| 2 |
+
GEMINI_API_KEY=AIzaSyCgOdLBoDEeG7vONzCZgm1l9dJFyIYrZOw
|
| 3 |
|
| 4 |
# CoinGecko API Key (Optional - for higher rate limits)
|
| 5 |
COINGECKO_API_KEY=your_coingecko_api_key_here
|
.gitignore
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
__pycache__/
|
| 2 |
+
*.pyc
|
| 3 |
+
*.pyo
|
| 4 |
+
*.pyd
|
| 5 |
+
.Python
|
| 6 |
+
env/
|
| 7 |
+
venv/
|
| 8 |
+
.venv/
|
| 9 |
+
pip-log.txt
|
| 10 |
+
pip-delete-this-directory.txt
|
| 11 |
+
.tox/
|
| 12 |
+
.coverage
|
| 13 |
+
.coverage.*
|
| 14 |
+
.cache
|
| 15 |
+
nosetests.xml
|
| 16 |
+
coverage.xml
|
| 17 |
+
*.cover
|
| 18 |
+
*.log
|
| 19 |
+
.git
|
| 20 |
+
.mypy_cache
|
| 21 |
+
.pytest_cache
|
| 22 |
+
.hypothesis
|
| 23 |
+
|
| 24 |
+
.DS_Store
|
| 25 |
+
.env
|
| 26 |
+
.flaskenv
|
| 27 |
+
*.env
|
| 28 |
+
|
| 29 |
+
gradio_queue.db
|
Dockerfile
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM python:3.11-slim
|
| 2 |
+
|
| 3 |
+
WORKDIR /app
|
| 4 |
+
|
| 5 |
+
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
| 6 |
+
|
| 7 |
+
COPY requirements.txt .
|
| 8 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
| 9 |
+
|
| 10 |
+
COPY . .
|
| 11 |
+
|
| 12 |
+
EXPOSE 7860
|
| 13 |
+
|
| 14 |
+
ENV PYTHONPATH=/app
|
| 15 |
+
ENV GRADIO_SERVER_NAME=0.0.0.0
|
| 16 |
+
ENV GRADIO_SERVER_PORT=7860
|
| 17 |
+
|
| 18 |
+
CMD ["python", "app.py"]
|
README.md
ADDED
|
@@ -0,0 +1,66 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Web3 Research Co-Pilot
|
| 2 |
+
|
| 3 |
+
AI-powered cryptocurrency research assistant built with LangChain and Gradio.
|
| 4 |
+
|
| 5 |
+
## Features
|
| 6 |
+
|
| 7 |
+
- **Real-time Market Analysis**: CoinGecko, DeFiLlama, Etherscan integration
|
| 8 |
+
- **AI Research Agent**: Powered by Google Gemini
|
| 9 |
+
- **Interactive Interface**: Modern Gradio UI
|
| 10 |
+
- **Data Visualization**: Price charts and market overviews
|
| 11 |
+
- **AIRAA Integration**: Webhook support for external platforms
|
| 12 |
+
|
| 13 |
+
## Quick Start
|
| 14 |
+
|
| 15 |
+
1. **Clone and Setup**
|
| 16 |
+
```bash
|
| 17 |
+
git clone <repository-url>
|
| 18 |
+
cd web3-research-agent
|
| 19 |
+
pip install -r requirements.txt
|
| 20 |
+
```
|
| 21 |
+
|
| 22 |
+
2. **Environment Configuration**
|
| 23 |
+
```bash
|
| 24 |
+
cp .env.example .env
|
| 25 |
+
# Edit .env with your API keys
|
| 26 |
+
```
|
| 27 |
+
|
| 28 |
+
3. **Run Application**
|
| 29 |
+
```bash
|
| 30 |
+
python app.py
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Required API Keys
|
| 34 |
+
|
| 35 |
+
- `GEMINI_API_KEY`: Google Gemini AI (required)
|
| 36 |
+
- `ETHERSCAN_API_KEY`: Ethereum blockchain data
|
| 37 |
+
- `COINGECKO_API_KEY`: Cryptocurrency market data (optional)
|
| 38 |
+
- `AIRAA_WEBHOOK_URL`: External integration (optional)
|
| 39 |
+
|
| 40 |
+
## Deployment
|
| 41 |
+
|
| 42 |
+
### Docker
|
| 43 |
+
```bash
|
| 44 |
+
docker build -t web3-research-agent .
|
| 45 |
+
docker run -p 7860:7860 --env-file .env web3-research-agent
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
### Hugging Face Spaces
|
| 49 |
+
Upload repository to HF Spaces with environment variables configured.
|
| 50 |
+
|
| 51 |
+
## Architecture
|
| 52 |
+
|
| 53 |
+
- **Agent**: LangChain-based research agent with memory
|
| 54 |
+
- **Tools**: Modular API integrations (CoinGecko, DeFiLlama, Etherscan)
|
| 55 |
+
- **UI**: Gradio interface with chat and visualization
|
| 56 |
+
- **Cache**: Optimized caching for API responses
|
| 57 |
+
- **Integration**: AIRAA webhook support
|
| 58 |
+
|
| 59 |
+
## Usage Examples
|
| 60 |
+
|
| 61 |
+
- "Bitcoin price analysis and market sentiment"
|
| 62 |
+
- "Top DeFi protocols by TVL"
|
| 63 |
+
- "Ethereum gas prices and network stats"
|
| 64 |
+
- "Compare BTC vs ETH performance"
|
| 65 |
+
|
| 66 |
+
Built with β€οΈ for Web3 research
|
app.py
CHANGED
|
@@ -1,312 +1,151 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
import os
|
| 3 |
-
import json
|
| 4 |
-
from src.enhanced_agent import EnhancedResearchAgent
|
| 5 |
-
from src.portfolio_analyzer import portfolio_analyzer
|
| 6 |
-
from src.visualizations import create_price_chart, create_market_overview, create_comparison_chart
|
| 7 |
-
from src.cache_manager import cache_manager
|
| 8 |
import asyncio
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
-
research_agent
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
return history + [["Please enter a research query.", ""]]
|
| 16 |
-
|
| 17 |
-
response = await research_agent.research_with_context(query)
|
| 18 |
-
return history + [[query, response]]
|
| 19 |
-
except Exception as e:
|
| 20 |
-
error_msg = f"Enhanced research failed: {str(e)}"
|
| 21 |
-
return history + [[query, error_msg]]
|
| 22 |
-
|
| 23 |
-
def research_query_sync(query, history):
|
| 24 |
-
return asyncio.run(process_research_query(query, history))
|
| 25 |
-
|
| 26 |
-
async def get_market_data():
|
| 27 |
-
try:
|
| 28 |
-
data = await research_agent.get_comprehensive_market_data()
|
| 29 |
-
chart = create_market_overview(data)
|
| 30 |
-
return chart
|
| 31 |
-
except Exception as e:
|
| 32 |
-
return f"Enhanced market data unavailable: {str(e)}"
|
| 33 |
-
|
| 34 |
-
def get_market_data_sync():
|
| 35 |
-
return asyncio.run(get_market_data())
|
| 36 |
-
|
| 37 |
-
async def get_price_chart(symbol):
|
| 38 |
-
try:
|
| 39 |
-
if not symbol.strip():
|
| 40 |
-
return "Please enter a cryptocurrency symbol"
|
| 41 |
-
|
| 42 |
-
data = await research_agent.get_price_history(symbol)
|
| 43 |
-
chart = create_price_chart(data, symbol)
|
| 44 |
-
return chart
|
| 45 |
-
except Exception as e:
|
| 46 |
-
return f"Chart generation failed: {str(e)}"
|
| 47 |
-
|
| 48 |
-
def get_price_chart_sync(symbol):
|
| 49 |
-
return asyncio.run(get_price_chart(symbol))
|
| 50 |
|
| 51 |
-
|
| 52 |
-
try:
|
| 53 |
-
if not portfolio_text.strip():
|
| 54 |
-
return "Please enter your portfolio holdings in JSON format"
|
| 55 |
-
|
| 56 |
-
holdings = json.loads(portfolio_text)
|
| 57 |
-
analysis = await portfolio_analyzer.analyze_portfolio(holdings)
|
| 58 |
-
|
| 59 |
-
result = f"π PORTFOLIO ANALYSIS\n\n"
|
| 60 |
-
result += f"π° Total Value: ${analysis['total_value']:,.2f}\n"
|
| 61 |
-
result += f"π 24h Change: ${analysis['change_24h']:+,.2f} ({analysis['change_24h_percentage']:+.2f}%)\n\n"
|
| 62 |
-
|
| 63 |
-
result += "π¦ ASSET ALLOCATION:\n"
|
| 64 |
-
for asset in analysis['asset_allocation'][:10]:
|
| 65 |
-
result += f"β’ {asset['name']} ({asset['symbol']}): {asset['percentage']:.1f}% (${asset['value']:,.2f})\n"
|
| 66 |
-
|
| 67 |
-
result += f"\nβ οΈ RISK ASSESSMENT:\n"
|
| 68 |
-
result += f"Overall Risk: {analysis['risk_metrics']['overall_risk']}\n"
|
| 69 |
-
result += f"Diversification Score: {analysis['risk_metrics']['diversification_score']}/10\n"
|
| 70 |
-
result += f"Largest Position: {analysis['risk_metrics']['largest_holding_percentage']:.1f}%\n\n"
|
| 71 |
-
|
| 72 |
-
result += "π‘ RECOMMENDATIONS:\n"
|
| 73 |
-
for i, rec in enumerate(analysis['recommendations'], 1):
|
| 74 |
-
result += f"{i}. {rec}\n"
|
| 75 |
-
|
| 76 |
-
return result
|
| 77 |
-
|
| 78 |
-
except json.JSONDecodeError:
|
| 79 |
-
return "β Invalid JSON format. Please use format: [{'symbol': 'BTC', 'amount': 1.0}, {'symbol': 'ETH', 'amount': 10.0}]"
|
| 80 |
-
except Exception as e:
|
| 81 |
-
return f"β Portfolio analysis failed: {str(e)}"
|
| 82 |
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
|
| 96 |
-
result += f"{i:2d}. {name} ({chain}): ${tvl/1e9:.2f}B TVL ({change:+.2f}%)\n"
|
| 97 |
-
|
| 98 |
-
if "top_yields" in data:
|
| 99 |
-
result += "\nπ° HIGH YIELD OPPORTUNITIES:\n"
|
| 100 |
-
for i, pool in enumerate(data["top_yields"][:5], 1):
|
| 101 |
-
symbol = pool.get("symbol", "Unknown")
|
| 102 |
-
apy = pool.get("apy", 0)
|
| 103 |
-
tvl = pool.get("tvlUsd", 0)
|
| 104 |
-
result += f"{i}. {symbol}: {apy:.2f}% APY (${tvl/1e6:.1f}M TVL)\n"
|
| 105 |
-
|
| 106 |
-
return result
|
| 107 |
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
|
| 132 |
-
|
| 133 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 134 |
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 138 |
with gr.Row():
|
| 139 |
-
with gr.Column(scale=
|
| 140 |
-
chatbot = gr.Chatbot(
|
| 141 |
-
value=[],
|
| 142 |
-
height=400,
|
| 143 |
-
elem_classes=["chat-container"],
|
| 144 |
-
show_label=False
|
| 145 |
-
)
|
| 146 |
|
| 147 |
with gr.Row():
|
| 148 |
query_input = gr.Textbox(
|
| 149 |
-
placeholder="Ask about crypto markets,
|
| 150 |
-
|
| 151 |
-
show_label=False
|
| 152 |
)
|
| 153 |
-
submit_btn = gr.Button("Research", variant="primary")
|
| 154 |
|
| 155 |
-
gr.
|
| 156 |
-
examples=[
|
| 157 |
-
"What's the current Bitcoin price and trend?",
|
| 158 |
-
"Compare Ethereum vs Solana DeFi ecosystems",
|
| 159 |
-
"Analyze top DeFi protocols and TVL trends",
|
| 160 |
-
"What are the trending coins and latest crypto news?",
|
| 161 |
-
"Show me high-yield DeFi opportunities with risk analysis"
|
| 162 |
-
],
|
| 163 |
-
inputs=query_input
|
| 164 |
-
)
|
| 165 |
|
| 166 |
with gr.Column(scale=1):
|
| 167 |
-
gr.Markdown("###
|
| 168 |
-
market_btn = gr.Button("Market Overview", size="sm")
|
| 169 |
-
market_output = gr.HTML()
|
| 170 |
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 184 |
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
inputs=
|
|
|
|
| 188 |
)
|
| 189 |
-
|
| 190 |
-
with gr.Tab("πΌ Portfolio Analysis"):
|
| 191 |
-
with gr.Row():
|
| 192 |
-
with gr.Column(scale=1):
|
| 193 |
-
portfolio_input = gr.Textbox(
|
| 194 |
-
label="Portfolio Holdings (JSON Format)",
|
| 195 |
-
placeholder='[{"symbol": "BTC", "amount": 1.0}, {"symbol": "ETH", "amount": 10.0}, {"symbol": "SOL", "amount": 50.0}]',
|
| 196 |
-
lines=5,
|
| 197 |
-
info="Enter your crypto holdings in JSON format with symbol and amount"
|
| 198 |
-
)
|
| 199 |
-
portfolio_btn = gr.Button("Analyze Portfolio", variant="primary")
|
| 200 |
-
|
| 201 |
-
with gr.Column(scale=2):
|
| 202 |
-
portfolio_output = gr.Textbox(
|
| 203 |
-
label="Portfolio Analysis Results",
|
| 204 |
-
lines=20,
|
| 205 |
-
show_copy_button=True,
|
| 206 |
-
interactive=False
|
| 207 |
-
)
|
| 208 |
|
| 209 |
-
|
| 210 |
-
|
| 211 |
-
|
| 212 |
-
|
| 213 |
-
],
|
| 214 |
-
inputs=portfolio_input
|
| 215 |
-
)
|
| 216 |
-
|
| 217 |
-
with gr.Tab("π¦ DeFi Analytics"):
|
| 218 |
-
defi_btn = gr.Button("Get DeFi Ecosystem Analysis", variant="primary", size="lg")
|
| 219 |
-
defi_output = gr.Textbox(
|
| 220 |
-
label="DeFi Analysis Results",
|
| 221 |
-
lines=25,
|
| 222 |
-
show_copy_button=True,
|
| 223 |
-
interactive=False,
|
| 224 |
-
info="Comprehensive DeFi protocol analysis with TVL data and yield opportunities"
|
| 225 |
)
|
| 226 |
-
|
| 227 |
-
with gr.Tab("βΉοΈ About"):
|
| 228 |
-
gr.Markdown("""
|
| 229 |
-
## π Enhanced Features
|
| 230 |
-
- **Multi-API Integration**: CoinGecko, CryptoCompare, and DeFiLlama data sources
|
| 231 |
-
- **AI-Powered Analysis**: Google Gemini 2.5 Flash with contextual market intelligence
|
| 232 |
-
- **DeFi Analytics**: Protocol TVL analysis, yield farming opportunities, and ecosystem insights
|
| 233 |
-
- **Portfolio Analysis**: Risk assessment, diversification scoring, and personalized recommendations
|
| 234 |
-
- **News Integration**: Real-time crypto news aggregation and sentiment analysis
|
| 235 |
-
- **Interactive Charts**: Advanced price visualizations with technical indicators
|
| 236 |
-
- **Smart Caching**: Optimized performance with intelligent data caching (TTL-based)
|
| 237 |
-
- **Rate Limiting**: Respectful API usage with automatic throttling
|
| 238 |
|
| 239 |
-
|
| 240 |
-
1. **Enhanced Research Chat**: Context-aware conversations with real-time market data integration
|
| 241 |
-
2. **Advanced Price Charts**: Interactive visualizations with 30-day historical data
|
| 242 |
-
3. **Portfolio Optimization**: Comprehensive portfolio analysis with risk metrics and recommendations
|
| 243 |
-
4. **DeFi Intelligence**: Protocol rankings, TVL trends, and high-yield opportunity identification
|
| 244 |
-
5. **Market Intelligence**: Global market metrics, trending assets, and breaking news analysis
|
| 245 |
|
| 246 |
-
|
| 247 |
-
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
|
| 251 |
-
- "Latest crypto news impact on altcoin market sentiment"
|
| 252 |
-
- "Which Layer 1 protocols have strongest DeFi adoption?"
|
| 253 |
|
| 254 |
-
|
| 255 |
-
|
| 256 |
-
|
| 257 |
-
|
| 258 |
-
|
| 259 |
-
|
| 260 |
-
|
| 261 |
-
submit_btn.click(
|
| 262 |
-
research_query_sync,
|
| 263 |
-
inputs=[query_input, chatbot],
|
| 264 |
-
outputs=chatbot
|
| 265 |
-
).then(lambda: "", outputs=query_input)
|
| 266 |
-
|
| 267 |
-
query_input.submit(
|
| 268 |
-
research_query_sync,
|
| 269 |
-
inputs=[query_input, chatbot],
|
| 270 |
-
outputs=chatbot
|
| 271 |
-
).then(lambda: "", outputs=query_input)
|
| 272 |
-
|
| 273 |
-
chart_btn.click(
|
| 274 |
-
get_price_chart_sync,
|
| 275 |
-
inputs=symbol_input,
|
| 276 |
-
outputs=chart_output
|
| 277 |
-
)
|
| 278 |
-
|
| 279 |
-
symbol_input.submit(
|
| 280 |
-
get_price_chart_sync,
|
| 281 |
-
inputs=symbol_input,
|
| 282 |
-
outputs=chart_output
|
| 283 |
-
)
|
| 284 |
-
|
| 285 |
-
market_btn.click(
|
| 286 |
-
get_market_data_sync,
|
| 287 |
-
outputs=market_output
|
| 288 |
-
)
|
| 289 |
-
|
| 290 |
-
clear_btn.click(
|
| 291 |
-
clear_cache,
|
| 292 |
-
outputs=clear_output
|
| 293 |
-
)
|
| 294 |
-
|
| 295 |
-
portfolio_btn.click(
|
| 296 |
-
analyze_portfolio_sync,
|
| 297 |
-
inputs=portfolio_input,
|
| 298 |
-
outputs=portfolio_output
|
| 299 |
-
)
|
| 300 |
-
|
| 301 |
-
defi_btn.click(
|
| 302 |
-
get_defi_analysis_sync,
|
| 303 |
-
outputs=defi_output
|
| 304 |
-
)
|
| 305 |
|
| 306 |
if __name__ == "__main__":
|
| 307 |
-
app
|
|
|
|
|
|
|
| 308 |
server_name="0.0.0.0",
|
| 309 |
-
server_port=
|
| 310 |
-
share=False
|
| 311 |
-
show_error=True
|
| 312 |
)
|
|
|
|
| 1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
import asyncio
|
| 3 |
+
import json
|
| 4 |
+
from datetime import datetime
|
| 5 |
+
from typing import List, Tuple
|
| 6 |
|
| 7 |
+
from src.agent.research_agent import Web3ResearchAgent
|
| 8 |
+
from src.api.airaa_integration import AIRAAIntegration
|
| 9 |
+
from src.visualizations import create_price_chart, create_market_overview
|
| 10 |
+
from src.utils.logger import get_logger
|
| 11 |
+
from src.utils.config import config
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
+
logger = get_logger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
+
class Web3CoPilotApp:
|
| 16 |
+
def __init__(self):
|
| 17 |
+
try:
|
| 18 |
+
self.agent = Web3ResearchAgent()
|
| 19 |
+
self.airaa = AIRAAIntegration()
|
| 20 |
+
except Exception as e:
|
| 21 |
+
logger.error(f"App initialization failed: {e}")
|
| 22 |
+
raise
|
| 23 |
+
|
| 24 |
+
async def process_query(self, query: str, history: List[Tuple[str, str]]):
|
| 25 |
+
if not query.strip():
|
| 26 |
+
yield history, ""
|
| 27 |
+
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
+
try:
|
| 30 |
+
history.append((query, "π Researching..."))
|
| 31 |
+
yield history, ""
|
| 32 |
+
|
| 33 |
+
result = await self.agent.research_query(query)
|
| 34 |
+
|
| 35 |
+
if result["success"]:
|
| 36 |
+
response = result["result"]
|
| 37 |
+
sources = ", ".join(result.get("sources", []))
|
| 38 |
+
response += f"\n\n---\nπ **Sources**: {sources}\nβ° **Generated**: {datetime.now().strftime('%H:%M:%S')}"
|
| 39 |
+
|
| 40 |
+
if config.AIRAA_WEBHOOK_URL:
|
| 41 |
+
asyncio.create_task(self.airaa.send_research_data(result))
|
| 42 |
+
else:
|
| 43 |
+
response = f"β Error: {result.get('error', 'Research failed')}"
|
| 44 |
+
|
| 45 |
+
history[-1] = (query, response)
|
| 46 |
+
yield history, ""
|
| 47 |
+
|
| 48 |
+
except Exception as e:
|
| 49 |
+
logger.error(f"Query error: {e}")
|
| 50 |
+
history[-1] = (query, f"β System error: {str(e)}")
|
| 51 |
+
yield history, ""
|
| 52 |
|
| 53 |
+
def get_chart_data(self, symbol: str):
|
| 54 |
+
try:
|
| 55 |
+
if not symbol.strip():
|
| 56 |
+
return "Please enter a symbol"
|
| 57 |
+
|
| 58 |
+
data = asyncio.run(self.agent.get_price_history(symbol))
|
| 59 |
+
return create_price_chart(data, symbol)
|
| 60 |
+
except Exception as e:
|
| 61 |
+
logger.error(f"Chart error: {e}")
|
| 62 |
+
return f"Chart unavailable: {str(e)}"
|
| 63 |
|
| 64 |
+
def get_market_overview(self):
|
| 65 |
+
try:
|
| 66 |
+
data = asyncio.run(self.agent.get_comprehensive_market_data())
|
| 67 |
+
return create_market_overview(data)
|
| 68 |
+
except Exception as e:
|
| 69 |
+
logger.error(f"Market overview error: {e}")
|
| 70 |
+
return f"Market data unavailable: {str(e)}"
|
| 71 |
+
|
| 72 |
+
def create_interface(self):
|
| 73 |
+
with gr.Blocks(title=config.UI_TITLE, theme=gr.themes.Soft()) as demo:
|
| 74 |
+
gr.Markdown(f"""
|
| 75 |
+
# π {config.UI_TITLE}
|
| 76 |
+
{config.UI_DESCRIPTION}
|
| 77 |
+
**Powered by**: Gemini AI β’ CoinGecko β’ DeFiLlama β’ Etherscan
|
| 78 |
+
""")
|
| 79 |
+
|
| 80 |
with gr.Row():
|
| 81 |
+
with gr.Column(scale=2):
|
| 82 |
+
chatbot = gr.Chatbot(label="Research Assistant", height=650)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
with gr.Row():
|
| 85 |
query_input = gr.Textbox(
|
| 86 |
+
placeholder="Ask about crypto markets, DeFi protocols, or on-chain data...",
|
| 87 |
+
label="Research Query", lines=2
|
|
|
|
| 88 |
)
|
| 89 |
+
submit_btn = gr.Button("π Research", variant="primary")
|
| 90 |
|
| 91 |
+
clear_btn = gr.Button("ποΈ Clear", variant="secondary")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 92 |
|
| 93 |
with gr.Column(scale=1):
|
| 94 |
+
gr.Markdown("### π‘ Example Queries")
|
|
|
|
|
|
|
| 95 |
|
| 96 |
+
examples = [
|
| 97 |
+
"Bitcoin price analysis",
|
| 98 |
+
"Top DeFi protocols by TVL",
|
| 99 |
+
"Ethereum vs Solana comparison",
|
| 100 |
+
"Trending cryptocurrencies",
|
| 101 |
+
"DeFi yield opportunities"
|
| 102 |
+
]
|
| 103 |
+
|
| 104 |
+
for example in examples:
|
| 105 |
+
gr.Button(example, size="sm").click(
|
| 106 |
+
lambda x=example: x, outputs=query_input
|
| 107 |
+
)
|
| 108 |
+
|
| 109 |
+
gr.Markdown("### π Visualizations")
|
| 110 |
+
chart_output = gr.Plot(label="Charts")
|
| 111 |
+
|
| 112 |
+
symbol_input = gr.Textbox(placeholder="BTC, ETH, SOL...", label="Chart Symbol")
|
| 113 |
+
chart_btn = gr.Button("π Generate Chart")
|
| 114 |
+
|
| 115 |
+
market_btn = gr.Button("π Market Overview")
|
| 116 |
|
| 117 |
+
submit_btn.click(
|
| 118 |
+
self.process_query,
|
| 119 |
+
inputs=[query_input, chatbot],
|
| 120 |
+
outputs=[chatbot, query_input]
|
| 121 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 122 |
|
| 123 |
+
query_input.submit(
|
| 124 |
+
self.process_query,
|
| 125 |
+
inputs=[query_input, chatbot],
|
| 126 |
+
outputs=[chatbot, query_input]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 127 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 128 |
|
| 129 |
+
clear_btn.click(lambda: ([], ""), outputs=[chatbot, query_input])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 130 |
|
| 131 |
+
chart_btn.click(
|
| 132 |
+
self.get_chart_data,
|
| 133 |
+
inputs=symbol_input,
|
| 134 |
+
outputs=chart_output
|
| 135 |
+
)
|
|
|
|
|
|
|
| 136 |
|
| 137 |
+
market_btn.click(
|
| 138 |
+
self.get_market_overview,
|
| 139 |
+
outputs=chart_output
|
| 140 |
+
)
|
| 141 |
+
|
| 142 |
+
return demo
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
|
| 144 |
if __name__ == "__main__":
|
| 145 |
+
app = Web3CoPilotApp()
|
| 146 |
+
interface = app.create_interface()
|
| 147 |
+
interface.launch(
|
| 148 |
server_name="0.0.0.0",
|
| 149 |
+
server_port=7860,
|
| 150 |
+
share=False
|
|
|
|
| 151 |
)
|
minimal_test.py
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import sys
|
| 2 |
+
import os
|
| 3 |
+
|
| 4 |
+
sys.path.insert(0, os.path.dirname(__file__))
|
| 5 |
+
|
| 6 |
+
def minimal_test():
|
| 7 |
+
try:
|
| 8 |
+
print("Testing minimal imports...")
|
| 9 |
+
|
| 10 |
+
from src.utils.config import config
|
| 11 |
+
print("β
Config imported")
|
| 12 |
+
|
| 13 |
+
from src.utils.logger import get_logger
|
| 14 |
+
print("β
Logger imported")
|
| 15 |
+
|
| 16 |
+
from src.tools.base_tool import BaseWeb3Tool
|
| 17 |
+
print("β
Base tool imported")
|
| 18 |
+
|
| 19 |
+
from src.tools.coingecko_tool import CoinGeckoTool
|
| 20 |
+
tool = CoinGeckoTool()
|
| 21 |
+
print("β
CoinGecko tool created")
|
| 22 |
+
|
| 23 |
+
from src.agent.research_agent import Web3ResearchAgent
|
| 24 |
+
print("β
Research agent imported")
|
| 25 |
+
|
| 26 |
+
print("π All core components working!")
|
| 27 |
+
return True
|
| 28 |
+
|
| 29 |
+
except Exception as e:
|
| 30 |
+
print(f"β Error: {e}")
|
| 31 |
+
import traceback
|
| 32 |
+
traceback.print_exc()
|
| 33 |
+
return False
|
| 34 |
+
|
| 35 |
+
if __name__ == "__main__":
|
| 36 |
+
minimal_test()
|
requirements.txt
CHANGED
|
@@ -1,7 +1,14 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
langchain
|
| 2 |
+
langchain-google-genai
|
| 3 |
+
langchain-community
|
| 4 |
+
gradio
|
| 5 |
+
aiohttp
|
| 6 |
+
tenacity
|
| 7 |
+
plotly
|
| 8 |
+
pandas
|
| 9 |
+
numpy
|
| 10 |
+
pydantic
|
| 11 |
+
python-dotenv
|
| 12 |
+
diskcache
|
| 13 |
+
google-generativeai
|
| 14 |
+
asyncio-throttle
|
run.py
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
|
| 3 |
+
"""
|
| 4 |
+
Web3 Research Co-Pilot Application
|
| 5 |
+
Complete production-ready crypto research assistant powered by AI
|
| 6 |
+
"""
|
| 7 |
+
|
| 8 |
+
import sys
|
| 9 |
+
import os
|
| 10 |
+
import asyncio
|
| 11 |
+
from pathlib import Path
|
| 12 |
+
|
| 13 |
+
# Add project root to path
|
| 14 |
+
project_root = Path(__file__).parent
|
| 15 |
+
sys.path.insert(0, str(project_root))
|
| 16 |
+
|
| 17 |
+
def main():
|
| 18 |
+
print("π Starting Web3 Research Co-Pilot...")
|
| 19 |
+
|
| 20 |
+
try:
|
| 21 |
+
from app import Web3CoPilotApp
|
| 22 |
+
|
| 23 |
+
app = Web3CoPilotApp()
|
| 24 |
+
interface = app.create_interface()
|
| 25 |
+
|
| 26 |
+
print("β
Application initialized successfully!")
|
| 27 |
+
print("π Launching web interface...")
|
| 28 |
+
print("π Local URL: http://localhost:7860")
|
| 29 |
+
|
| 30 |
+
interface.launch(
|
| 31 |
+
server_name="0.0.0.0",
|
| 32 |
+
server_port=7860,
|
| 33 |
+
share=False,
|
| 34 |
+
show_api=False,
|
| 35 |
+
quiet=False
|
| 36 |
+
)
|
| 37 |
+
|
| 38 |
+
except ImportError as e:
|
| 39 |
+
print(f"β Import error: {e}")
|
| 40 |
+
print("Please install dependencies: pip install -r requirements.txt")
|
| 41 |
+
sys.exit(1)
|
| 42 |
+
except Exception as e:
|
| 43 |
+
print(f"β Application error: {e}")
|
| 44 |
+
sys.exit(1)
|
| 45 |
+
|
| 46 |
+
if __name__ == "__main__":
|
| 47 |
+
main()
|
src/agent/__init__.py
ADDED
|
File without changes
|
src/agent/query_planner.py
ADDED
|
@@ -0,0 +1,89 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any, List
|
| 2 |
+
import asyncio
|
| 3 |
+
|
| 4 |
+
class QueryPlanner:
|
| 5 |
+
def __init__(self, llm):
|
| 6 |
+
self.llm = llm
|
| 7 |
+
|
| 8 |
+
async def plan_research(self, query: str) -> Dict[str, Any]:
|
| 9 |
+
try:
|
| 10 |
+
planning_prompt = f"""
|
| 11 |
+
Analyze this research query and create a structured plan:
|
| 12 |
+
Query: {query}
|
| 13 |
+
|
| 14 |
+
Determine:
|
| 15 |
+
1. What type of analysis is needed (price, market, defi, comparison, etc.)
|
| 16 |
+
2. Which data sources would be most relevant
|
| 17 |
+
3. What specific steps should be taken
|
| 18 |
+
4. Priority focus area
|
| 19 |
+
|
| 20 |
+
Respond in JSON format with keys: type, steps, priority, data_sources
|
| 21 |
+
"""
|
| 22 |
+
|
| 23 |
+
response = await asyncio.to_thread(
|
| 24 |
+
self.llm.invoke,
|
| 25 |
+
planning_prompt
|
| 26 |
+
)
|
| 27 |
+
|
| 28 |
+
# Simple categorization based on keywords
|
| 29 |
+
query_lower = query.lower()
|
| 30 |
+
plan = {
|
| 31 |
+
"type": self._categorize_query(query_lower),
|
| 32 |
+
"steps": self._generate_steps(query_lower),
|
| 33 |
+
"priority": self._determine_priority(query_lower),
|
| 34 |
+
"data_sources": self._identify_sources(query_lower)
|
| 35 |
+
}
|
| 36 |
+
|
| 37 |
+
return plan
|
| 38 |
+
|
| 39 |
+
except Exception:
|
| 40 |
+
return {
|
| 41 |
+
"type": "general",
|
| 42 |
+
"steps": ["Analyze query", "Gather data", "Provide insights"],
|
| 43 |
+
"priority": "general analysis",
|
| 44 |
+
"data_sources": ["coingecko", "defillama"]
|
| 45 |
+
}
|
| 46 |
+
|
| 47 |
+
def _categorize_query(self, query: str) -> str:
|
| 48 |
+
if any(word in query for word in ["price", "chart", "value"]):
|
| 49 |
+
return "price_analysis"
|
| 50 |
+
elif any(word in query for word in ["defi", "tvl", "protocol", "yield"]):
|
| 51 |
+
return "defi_analysis"
|
| 52 |
+
elif any(word in query for word in ["compare", "vs", "versus"]):
|
| 53 |
+
return "comparison"
|
| 54 |
+
elif any(word in query for word in ["market", "overview", "trending"]):
|
| 55 |
+
return "market_overview"
|
| 56 |
+
else:
|
| 57 |
+
return "general"
|
| 58 |
+
|
| 59 |
+
def _generate_steps(self, query: str) -> List[str]:
|
| 60 |
+
steps = ["Gather relevant data"]
|
| 61 |
+
|
| 62 |
+
if "price" in query:
|
| 63 |
+
steps.extend(["Get current price data", "Analyze price trends"])
|
| 64 |
+
if "defi" in query:
|
| 65 |
+
steps.extend(["Fetch DeFi protocol data", "Analyze TVL trends"])
|
| 66 |
+
if any(word in query for word in ["compare", "vs"]):
|
| 67 |
+
steps.append("Perform comparative analysis")
|
| 68 |
+
|
| 69 |
+
steps.append("Synthesize insights and recommendations")
|
| 70 |
+
return steps
|
| 71 |
+
|
| 72 |
+
def _determine_priority(self, query: str) -> str:
|
| 73 |
+
if "urgent" in query or "now" in query:
|
| 74 |
+
return "high"
|
| 75 |
+
elif "overview" in query:
|
| 76 |
+
return "comprehensive"
|
| 77 |
+
else:
|
| 78 |
+
return "standard"
|
| 79 |
+
|
| 80 |
+
def _identify_sources(self, query: str) -> List[str]:
|
| 81 |
+
sources = []
|
| 82 |
+
if any(word in query for word in ["price", "market", "coin", "token"]):
|
| 83 |
+
sources.append("coingecko")
|
| 84 |
+
if any(word in query for word in ["defi", "tvl", "protocol"]):
|
| 85 |
+
sources.append("defillama")
|
| 86 |
+
if any(word in query for word in ["transaction", "address", "gas"]):
|
| 87 |
+
sources.append("etherscan")
|
| 88 |
+
|
| 89 |
+
return sources if sources else ["coingecko", "defillama"]
|
src/agent/research_agent.py
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from langchain.agents import AgentExecutor, create_tool_calling_agent
|
| 2 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 3 |
+
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
|
| 4 |
+
from langchain.memory import ConversationBufferWindowMemory
|
| 5 |
+
from typing import List, Dict, Any
|
| 6 |
+
import asyncio
|
| 7 |
+
from datetime import datetime
|
| 8 |
+
|
| 9 |
+
from src.tools.coingecko_tool import CoinGeckoTool
|
| 10 |
+
from src.tools.defillama_tool import DeFiLlamaTool
|
| 11 |
+
from src.tools.etherscan_tool import EtherscanTool
|
| 12 |
+
from src.agent.query_planner import QueryPlanner
|
| 13 |
+
from src.config import config
|
| 14 |
+
from src.utils.logger import get_logger
|
| 15 |
+
|
| 16 |
+
logger = get_logger(__name__)
|
| 17 |
+
|
| 18 |
+
class Web3ResearchAgent:
|
| 19 |
+
def __init__(self):
|
| 20 |
+
try:
|
| 21 |
+
if not config.GEMINI_API_KEY:
|
| 22 |
+
raise ValueError("GEMINI_API_KEY not configured")
|
| 23 |
+
|
| 24 |
+
self.llm = ChatGoogleGenerativeAI(
|
| 25 |
+
model="gemini-1.5-flash",
|
| 26 |
+
google_api_key=config.GEMINI_API_KEY,
|
| 27 |
+
temperature=0.1,
|
| 28 |
+
max_tokens=2048
|
| 29 |
+
)
|
| 30 |
+
|
| 31 |
+
self.tools = [CoinGeckoTool(), DeFiLlamaTool(), EtherscanTool()]
|
| 32 |
+
self.query_planner = QueryPlanner(self.llm)
|
| 33 |
+
self.memory = ConversationBufferWindowMemory(
|
| 34 |
+
memory_key="chat_history", return_messages=True, k=10
|
| 35 |
+
)
|
| 36 |
+
|
| 37 |
+
self.agent = self._create_agent()
|
| 38 |
+
self.executor = AgentExecutor(
|
| 39 |
+
agent=self.agent, tools=self.tools, memory=self.memory,
|
| 40 |
+
verbose=False, max_iterations=5, handle_parsing_errors=True
|
| 41 |
+
)
|
| 42 |
+
except Exception as e:
|
| 43 |
+
logger.error(f"Agent init failed: {e}")
|
| 44 |
+
raise
|
| 45 |
+
|
| 46 |
+
def _create_agent(self):
|
| 47 |
+
prompt = ChatPromptTemplate.from_messages([
|
| 48 |
+
("system", """You are an expert Web3 research assistant. Use available tools to provide accurate,
|
| 49 |
+
data-driven insights about cryptocurrency markets, DeFi protocols, and blockchain data.
|
| 50 |
+
|
| 51 |
+
Format responses with clear sections, emojis, and actionable insights."""),
|
| 52 |
+
MessagesPlaceholder("chat_history"),
|
| 53 |
+
("human", "{input}"),
|
| 54 |
+
MessagesPlaceholder("agent_scratchpad")
|
| 55 |
+
])
|
| 56 |
+
|
| 57 |
+
return create_tool_calling_agent(self.llm, self.tools, prompt)
|
| 58 |
+
|
| 59 |
+
async def research_query(self, query: str) -> Dict[str, Any]:
|
| 60 |
+
try:
|
| 61 |
+
logger.info(f"Processing: {query}")
|
| 62 |
+
|
| 63 |
+
research_plan = await self.query_planner.plan_research(query)
|
| 64 |
+
|
| 65 |
+
enhanced_query = f"""
|
| 66 |
+
Research Query: {query}
|
| 67 |
+
Research Plan: {research_plan.get('steps', [])}
|
| 68 |
+
Priority: {research_plan.get('priority', 'general')}
|
| 69 |
+
|
| 70 |
+
Execute systematic research and provide comprehensive analysis.
|
| 71 |
+
"""
|
| 72 |
+
|
| 73 |
+
result = await asyncio.to_thread(
|
| 74 |
+
self.executor.invoke, {"input": enhanced_query}
|
| 75 |
+
)
|
| 76 |
+
|
| 77 |
+
return {
|
| 78 |
+
"success": True,
|
| 79 |
+
"query": query,
|
| 80 |
+
"research_plan": research_plan,
|
| 81 |
+
"result": result.get("output", "No response"),
|
| 82 |
+
"sources": self._extract_sources(result.get("output", "")),
|
| 83 |
+
"metadata": {
|
| 84 |
+
"tools_used": [tool.name for tool in self.tools],
|
| 85 |
+
"timestamp": datetime.now().isoformat()
|
| 86 |
+
}
|
| 87 |
+
}
|
| 88 |
+
|
| 89 |
+
except Exception as e:
|
| 90 |
+
logger.error(f"Research error: {e}")
|
| 91 |
+
return {
|
| 92 |
+
"success": False,
|
| 93 |
+
"query": query,
|
| 94 |
+
"error": str(e),
|
| 95 |
+
"metadata": {"timestamp": datetime.now().isoformat()}
|
| 96 |
+
}
|
| 97 |
+
|
| 98 |
+
async def get_price_history(self, symbol: str, days: int = 30) -> Dict[str, Any]:
|
| 99 |
+
try:
|
| 100 |
+
coingecko_tool = next(t for t in self.tools if isinstance(t, CoinGeckoTool))
|
| 101 |
+
return await coingecko_tool._arun(symbol, {"type": "price_history", "days": days})
|
| 102 |
+
except Exception as e:
|
| 103 |
+
logger.error(f"Price history error: {e}")
|
| 104 |
+
return {}
|
| 105 |
+
|
| 106 |
+
async def get_comprehensive_market_data(self) -> Dict[str, Any]:
|
| 107 |
+
try:
|
| 108 |
+
tasks = []
|
| 109 |
+
for tool in self.tools:
|
| 110 |
+
if isinstance(tool, CoinGeckoTool):
|
| 111 |
+
tasks.append(tool._arun("", {"type": "market_overview"}))
|
| 112 |
+
elif isinstance(tool, DeFiLlamaTool):
|
| 113 |
+
tasks.append(tool._arun("", {"type": "tvl_overview"}))
|
| 114 |
+
|
| 115 |
+
results = await asyncio.gather(*tasks, return_exceptions=True)
|
| 116 |
+
|
| 117 |
+
data = {}
|
| 118 |
+
for i, result in enumerate(results):
|
| 119 |
+
if not isinstance(result, Exception):
|
| 120 |
+
if i == 0:
|
| 121 |
+
data["market"] = result
|
| 122 |
+
elif i == 1:
|
| 123 |
+
data["defi"] = result
|
| 124 |
+
|
| 125 |
+
return data
|
| 126 |
+
except Exception as e:
|
| 127 |
+
logger.error(f"Market data error: {e}")
|
| 128 |
+
return {}
|
| 129 |
+
|
| 130 |
+
def _extract_sources(self, result_text: str) -> List[str]:
|
| 131 |
+
sources = []
|
| 132 |
+
if "CoinGecko" in result_text or "coingecko" in result_text.lower():
|
| 133 |
+
sources.append("CoinGecko API")
|
| 134 |
+
if "DeFiLlama" in result_text or "defillama" in result_text.lower():
|
| 135 |
+
sources.append("DeFiLlama API")
|
| 136 |
+
if "Etherscan" in result_text or "etherscan" in result_text.lower():
|
| 137 |
+
sources.append("Etherscan API")
|
| 138 |
+
return sources
|
src/api/__init__.py
ADDED
|
File without changes
|
src/api/airaa_integration.py
ADDED
|
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import aiohttp
|
| 2 |
+
import re
|
| 3 |
+
from typing import Dict, Any, List
|
| 4 |
+
from src.config import config
|
| 5 |
+
from src.utils.logger import get_logger
|
| 6 |
+
|
| 7 |
+
logger = get_logger(__name__)
|
| 8 |
+
|
| 9 |
+
class AIRAAIntegration:
|
| 10 |
+
def __init__(self):
|
| 11 |
+
self.webhook_url = config.AIRAA_WEBHOOK_URL
|
| 12 |
+
self.api_key = config.AIRAA_API_KEY
|
| 13 |
+
self.enabled = bool(self.webhook_url)
|
| 14 |
+
|
| 15 |
+
async def send_research_data(self, research_result: Dict[str, Any]) -> bool:
|
| 16 |
+
if not self.enabled:
|
| 17 |
+
return False
|
| 18 |
+
|
| 19 |
+
try:
|
| 20 |
+
payload = self._format_for_airaa(research_result)
|
| 21 |
+
|
| 22 |
+
headers = {"Content-Type": "application/json"}
|
| 23 |
+
if self.api_key:
|
| 24 |
+
headers["Authorization"] = f"Bearer {self.api_key}"
|
| 25 |
+
|
| 26 |
+
timeout = aiohttp.ClientTimeout(total=30)
|
| 27 |
+
async with aiohttp.ClientSession(timeout=timeout) as session:
|
| 28 |
+
async with session.post(
|
| 29 |
+
self.webhook_url, json=payload, headers=headers
|
| 30 |
+
) as response:
|
| 31 |
+
success = response.status == 200
|
| 32 |
+
if success:
|
| 33 |
+
logger.info("Data sent to AIRAA successfully")
|
| 34 |
+
else:
|
| 35 |
+
logger.warning(f"AIRAA webhook returned {response.status}")
|
| 36 |
+
return success
|
| 37 |
+
|
| 38 |
+
except Exception as e:
|
| 39 |
+
logger.error(f"AIRAA integration failed: {e}")
|
| 40 |
+
return False
|
| 41 |
+
|
| 42 |
+
def _format_for_airaa(self, result: Dict[str, Any]) -> Dict[str, Any]:
|
| 43 |
+
return {
|
| 44 |
+
"source": "web3-research-copilot",
|
| 45 |
+
"timestamp": result["metadata"]["timestamp"],
|
| 46 |
+
"query": result["query"],
|
| 47 |
+
"research_plan": result.get("research_plan", {}),
|
| 48 |
+
"findings": result["result"],
|
| 49 |
+
"data_sources": result["sources"],
|
| 50 |
+
"confidence_score": self._calculate_confidence(result),
|
| 51 |
+
"tags": self._extract_tags(result["query"]),
|
| 52 |
+
"structured_data": self._extract_structured_data(result["result"])
|
| 53 |
+
}
|
| 54 |
+
|
| 55 |
+
def _calculate_confidence(self, result: Dict[str, Any]) -> float:
|
| 56 |
+
base_score = 0.7
|
| 57 |
+
source_boost = min(len(result.get("sources", [])) * 0.1, 0.3)
|
| 58 |
+
error_penalty = 0.3 if not result.get("success", True) else 0
|
| 59 |
+
return max(0.0, min(1.0, base_score + source_boost - error_penalty))
|
| 60 |
+
|
| 61 |
+
def _extract_tags(self, query: str) -> List[str]:
|
| 62 |
+
tags = []
|
| 63 |
+
query_lower = query.lower()
|
| 64 |
+
|
| 65 |
+
token_patterns = {
|
| 66 |
+
"bitcoin": ["bitcoin", "btc"],
|
| 67 |
+
"ethereum": ["ethereum", "eth"],
|
| 68 |
+
"defi": ["defi", "defillama", "protocol", "tvl"],
|
| 69 |
+
"market-analysis": ["price", "market", "analysis"],
|
| 70 |
+
"trading-volume": ["volume", "trading"]
|
| 71 |
+
}
|
| 72 |
+
|
| 73 |
+
for tag, patterns in token_patterns.items():
|
| 74 |
+
if any(pattern in query_lower for pattern in patterns):
|
| 75 |
+
tags.append(tag)
|
| 76 |
+
|
| 77 |
+
return tags
|
| 78 |
+
|
| 79 |
+
def _extract_structured_data(self, result_text: str) -> Dict[str, Any]:
|
| 80 |
+
structured = {}
|
| 81 |
+
|
| 82 |
+
price_pattern = r'\$([0-9,]+\.?[0-9]*)'
|
| 83 |
+
percentage_pattern = r'([+-]?[0-9]+\.?[0-9]*)%'
|
| 84 |
+
|
| 85 |
+
prices = re.findall(price_pattern, result_text)
|
| 86 |
+
percentages = re.findall(percentage_pattern, result_text)
|
| 87 |
+
|
| 88 |
+
if prices:
|
| 89 |
+
structured["prices"] = [float(p.replace(',', '')) for p in prices[:5]]
|
| 90 |
+
if percentages:
|
| 91 |
+
structured["percentages"] = [float(p) for p in percentages[:5]]
|
| 92 |
+
|
| 93 |
+
return structured
|
src/config.py
CHANGED
|
@@ -7,6 +7,7 @@ class Config:
|
|
| 7 |
GEMINI_API_KEY: str = os.getenv("GEMINI_API_KEY", "")
|
| 8 |
COINGECKO_API_KEY: Optional[str] = os.getenv("COINGECKO_API_KEY")
|
| 9 |
CRYPTOCOMPARE_API_KEY: Optional[str] = os.getenv("CRYPTOCOMPARE_API_KEY")
|
|
|
|
| 10 |
|
| 11 |
COINGECKO_BASE_URL: str = "https://api.coingecko.com/api/v3"
|
| 12 |
CRYPTOCOMPARE_BASE_URL: str = "https://min-api.cryptocompare.com/data"
|
|
@@ -15,5 +16,11 @@ class Config:
|
|
| 15 |
RATE_LIMIT_DELAY: float = 2.0
|
| 16 |
MAX_RETRIES: int = 3
|
| 17 |
REQUEST_TIMEOUT: int = 30
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
config = Config()
|
|
|
|
| 7 |
GEMINI_API_KEY: str = os.getenv("GEMINI_API_KEY", "")
|
| 8 |
COINGECKO_API_KEY: Optional[str] = os.getenv("COINGECKO_API_KEY")
|
| 9 |
CRYPTOCOMPARE_API_KEY: Optional[str] = os.getenv("CRYPTOCOMPARE_API_KEY")
|
| 10 |
+
ETHERSCAN_API_KEY: str = os.getenv("ETHERSCAN_API_KEY", "")
|
| 11 |
|
| 12 |
COINGECKO_BASE_URL: str = "https://api.coingecko.com/api/v3"
|
| 13 |
CRYPTOCOMPARE_BASE_URL: str = "https://min-api.cryptocompare.com/data"
|
|
|
|
| 16 |
RATE_LIMIT_DELAY: float = 2.0
|
| 17 |
MAX_RETRIES: int = 3
|
| 18 |
REQUEST_TIMEOUT: int = 30
|
| 19 |
+
|
| 20 |
+
UI_TITLE: str = "Web3 Research Co-Pilot"
|
| 21 |
+
UI_DESCRIPTION: str = "AI-powered crypto research assistant"
|
| 22 |
+
|
| 23 |
+
AIRAA_WEBHOOK_URL: Optional[str] = os.getenv("AIRAA_WEBHOOK_URL")
|
| 24 |
+
AIRAA_API_KEY: Optional[str] = os.getenv("AIRAA_API_KEY")
|
| 25 |
|
| 26 |
config = Config()
|
src/tools/__init__.py
ADDED
|
File without changes
|
src/tools/base_tool.py
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from abc import ABC, abstractmethod
|
| 2 |
+
from typing import Dict, Any, Optional
|
| 3 |
+
from langchain.tools import BaseTool
|
| 4 |
+
from pydantic import BaseModel, Field, PrivateAttr
|
| 5 |
+
import asyncio
|
| 6 |
+
import aiohttp
|
| 7 |
+
from tenacity import retry, stop_after_attempt, wait_exponential
|
| 8 |
+
from src.utils.logger import get_logger
|
| 9 |
+
|
| 10 |
+
logger = get_logger(__name__)
|
| 11 |
+
|
| 12 |
+
class Web3ToolInput(BaseModel):
|
| 13 |
+
query: str = Field(description="Search query or parameter")
|
| 14 |
+
filters: Optional[Dict[str, Any]] = Field(default=None, description="Additional filters")
|
| 15 |
+
|
| 16 |
+
class BaseWeb3Tool(BaseTool, ABC):
|
| 17 |
+
name: str = "base_web3_tool"
|
| 18 |
+
description: str = "Base Web3 tool"
|
| 19 |
+
args_schema: type[BaseModel] = Web3ToolInput
|
| 20 |
+
|
| 21 |
+
_session: Optional[aiohttp.ClientSession] = PrivateAttr(default=None)
|
| 22 |
+
|
| 23 |
+
def __init__(self, **kwargs):
|
| 24 |
+
super().__init__(**kwargs)
|
| 25 |
+
self._session = None
|
| 26 |
+
|
| 27 |
+
async def get_session(self):
|
| 28 |
+
if not self._session:
|
| 29 |
+
timeout = aiohttp.ClientTimeout(total=30)
|
| 30 |
+
self._session = aiohttp.ClientSession(timeout=timeout)
|
| 31 |
+
return self._session
|
| 32 |
+
|
| 33 |
+
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=2, max=8))
|
| 34 |
+
async def make_request(self, url: str, params: Dict[str, Any] = None) -> Dict[str, Any]:
|
| 35 |
+
session = await self.get_session()
|
| 36 |
+
try:
|
| 37 |
+
async with session.get(url, params=params or {}) as response:
|
| 38 |
+
if response.status == 200:
|
| 39 |
+
return await response.json()
|
| 40 |
+
elif response.status == 429:
|
| 41 |
+
await asyncio.sleep(2)
|
| 42 |
+
raise aiohttp.ClientResponseError(
|
| 43 |
+
request_info=response.request_info,
|
| 44 |
+
history=response.history,
|
| 45 |
+
status=response.status
|
| 46 |
+
)
|
| 47 |
+
else:
|
| 48 |
+
response.raise_for_status()
|
| 49 |
+
except Exception as e:
|
| 50 |
+
logger.error(f"Request failed: {e}")
|
| 51 |
+
raise
|
| 52 |
+
|
| 53 |
+
def _run(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
|
| 54 |
+
return asyncio.run(self._arun(query, filters))
|
| 55 |
+
|
| 56 |
+
@abstractmethod
|
| 57 |
+
async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
|
| 58 |
+
pass
|
| 59 |
+
|
| 60 |
+
async def cleanup(self):
|
| 61 |
+
if self._session:
|
| 62 |
+
await self._session.close()
|
| 63 |
+
if self.session:
|
| 64 |
+
await self.session.close()
|
src/tools/coingecko_tool.py
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any, Optional
|
| 2 |
+
from pydantic import BaseModel, PrivateAttr
|
| 3 |
+
from src.tools.base_tool import BaseWeb3Tool, Web3ToolInput
|
| 4 |
+
from src.utils.config import config
|
| 5 |
+
|
| 6 |
+
class CoinGeckoTool(BaseWeb3Tool):
|
| 7 |
+
name: str = "coingecko_data"
|
| 8 |
+
description: str = """Get cryptocurrency price, volume, market cap and trend data from CoinGecko.
|
| 9 |
+
Useful for: price analysis, market rankings, volume trends, price changes.
|
| 10 |
+
Input: cryptocurrency name/symbol (bitcoin, ethereum, BTC, ETH) or market query."""
|
| 11 |
+
args_schema: type[BaseModel] = Web3ToolInput
|
| 12 |
+
|
| 13 |
+
_base_url: str = PrivateAttr(default="https://api.coingecko.com/api/v3")
|
| 14 |
+
_symbol_map: Dict[str, str] = PrivateAttr(default_factory=lambda: {
|
| 15 |
+
"btc": "bitcoin", "eth": "ethereum", "sol": "solana", "ada": "cardano",
|
| 16 |
+
"dot": "polkadot", "bnb": "binancecoin", "usdc": "usd-coin",
|
| 17 |
+
"usdt": "tether", "xrp": "ripple", "avax": "avalanche-2",
|
| 18 |
+
"link": "chainlink", "matic": "matic-network", "uni": "uniswap"
|
| 19 |
+
})
|
| 20 |
+
|
| 21 |
+
def __init__(self):
|
| 22 |
+
super().__init__()
|
| 23 |
+
|
| 24 |
+
async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
|
| 25 |
+
try:
|
| 26 |
+
filters = filters or {}
|
| 27 |
+
|
| 28 |
+
if filters.get("type") == "trending":
|
| 29 |
+
return await self._get_trending()
|
| 30 |
+
elif filters.get("type") == "market_overview":
|
| 31 |
+
return await self._get_market_overview()
|
| 32 |
+
elif filters.get("type") == "price_history":
|
| 33 |
+
return await self._get_price_history(query, filters.get("days", 30))
|
| 34 |
+
else:
|
| 35 |
+
return await self._get_coin_data(query)
|
| 36 |
+
|
| 37 |
+
except Exception as e:
|
| 38 |
+
return f"CoinGecko error: {str(e)}"
|
| 39 |
+
async def _get_trending(self) -> str:
|
| 40 |
+
data = await self.make_request(f"{self._base_url}/search/trending")
|
| 41 |
+
data = await self.make_request(f"{self.base_url}/search/trending")
|
| 42 |
+
|
| 43 |
+
trending = data.get("coins", [])[:5]
|
| 44 |
+
result = "π₯ **Trending Cryptocurrencies:**\n\n"
|
| 45 |
+
|
| 46 |
+
for i, coin in enumerate(trending, 1):
|
| 47 |
+
item = coin.get("item", {})
|
| 48 |
+
name = item.get("name", "Unknown")
|
| 49 |
+
symbol = item.get("symbol", "").upper()
|
| 50 |
+
rank = item.get("market_cap_rank", "N/A")
|
| 51 |
+
result += f"{i}. **{name} ({symbol})** - Rank #{rank}\n"
|
| 52 |
+
|
| 53 |
+
return result
|
| 54 |
+
|
| 55 |
+
async def _get_market_overview(self) -> str:
|
| 56 |
+
data = await self.make_request(f"{self._base_url}/coins/markets", params)10, "page": 1}
|
| 57 |
+
data = await self.make_request(f"{self.base_url}/coins/markets", params)
|
| 58 |
+
|
| 59 |
+
result = "π **Top Cryptocurrencies by Market Cap:**\n\n"
|
| 60 |
+
|
| 61 |
+
for coin in data[:10]:
|
| 62 |
+
name = coin.get("name", "Unknown")
|
| 63 |
+
symbol = coin.get("symbol", "").upper()
|
| 64 |
+
price = coin.get("current_price", 0)
|
| 65 |
+
change = coin.get("price_change_percentage_24h", 0)
|
| 66 |
+
mcap = coin.get("market_cap", 0)
|
| 67 |
+
|
| 68 |
+
emoji = "π" if change >= 0 else "π"
|
| 69 |
+
result += f"{emoji} **{name} ({symbol})**: ${price:,.4f} ({change:+.2f}%) | MCap: ${mcap/1e9:.2f}B\n"
|
| 70 |
+
|
| 71 |
+
return result
|
| 72 |
+
|
| 73 |
+
coin_id = self._symbol_map.get(query.lower(), query.lower())
|
| 74 |
+
coin_id = self.symbol_map.get(query.lower(), query.lower())
|
| 75 |
+
|
| 76 |
+
params = {
|
| 77 |
+
"ids": coin_id,
|
| 78 |
+
"vs_currencies": "usd",
|
| 79 |
+
"include_24hr_change": "true",
|
| 80 |
+
"include_24hr_vol": "true",
|
| 81 |
+
"include_market_cap": "true"
|
| 82 |
+
}
|
| 83 |
+
data = await self.make_request(f"{self._base_url}/simple/price", params)
|
| 84 |
+
data = await self.make_request(f"{self.base_url}/simple/price", params)
|
| 85 |
+
|
| 86 |
+
if coin_id not in data:
|
| 87 |
+
return f"No data found for {query}"
|
| 88 |
+
|
| 89 |
+
coin_data = data[coin_id]
|
| 90 |
+
price = coin_data.get("usd", 0)
|
| 91 |
+
change = coin_data.get("usd_24h_change", 0)
|
| 92 |
+
volume = coin_data.get("usd_24h_vol", 0)
|
| 93 |
+
mcap = coin_data.get("usd_market_cap", 0)
|
| 94 |
+
|
| 95 |
+
emoji = "π" if change >= 0 else "π"
|
| 96 |
+
|
| 97 |
+
result = f"π° **{query.upper()} Market Data:**\n\n"
|
| 98 |
+
result += f"{emoji} **Price**: ${price:,.4f}\n"
|
| 99 |
+
result += f"π **24h Change**: {change:+.2f}%\n"
|
| 100 |
+
result += f"π **24h Volume**: ${volume:,.0f}\n"
|
| 101 |
+
result += f"π¦ **Market Cap**: ${mcap:,.0f}\n"
|
| 102 |
+
|
| 103 |
+
return result
|
| 104 |
+
|
| 105 |
+
coin_id = self._symbol_map.get(symbol.lower(), symbol.lower())
|
| 106 |
+
|
| 107 |
+
params = {"vs_currency": "usd", "days": days}
|
| 108 |
+
data = await self.make_request(f"{self._base_url}/coins/{coin_id}/market_chart", params)
|
| 109 |
+
data = await self.make_request(f"{self.base_url}/coins/{coin_id}/market_chart", params)
|
| 110 |
+
|
| 111 |
+
return {
|
| 112 |
+
"symbol": symbol.upper(),
|
| 113 |
+
"prices": data.get("prices", []),
|
| 114 |
+
"volumes": data.get("total_volumes", []),
|
| 115 |
+
"market_caps": data.get("market_caps", [])
|
| 116 |
+
}
|
src/tools/defillama_tool.py
ADDED
|
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any, Optional
|
| 2 |
+
from pydantic import BaseModel, PrivateAttr
|
| 3 |
+
from src.tools.base_tool import BaseWeb3Tool, Web3ToolInput
|
| 4 |
+
|
| 5 |
+
class DeFiLlamaTool(BaseWeb3Tool):
|
| 6 |
+
name: str = "defillama_data"
|
| 7 |
+
description: str = """Get DeFi protocol data, TVL, and yields from DeFiLlama.
|
| 8 |
+
Useful for: DeFi analysis, protocol rankings, TVL trends, yield farming data.
|
| 9 |
+
Input: protocol name or general DeFi query."""
|
| 10 |
+
args_schema: type[BaseModel] = Web3ToolInput
|
| 11 |
+
|
| 12 |
+
_base_url: str = PrivateAttr(default="https://api.llama.fi")
|
| 13 |
+
|
| 14 |
+
def __init__(self):
|
| 15 |
+
super().__init__()
|
| 16 |
+
|
| 17 |
+
async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
|
| 18 |
+
try:
|
| 19 |
+
filters = filters or {}
|
| 20 |
+
|
| 21 |
+
if filters.get("type") == "tvl_overview":
|
| 22 |
+
return await self._get_tvl_overview()
|
| 23 |
+
elif filters.get("type") == "protocol_data":
|
| 24 |
+
return await self._get_protocol_data(query)
|
| 25 |
+
elif query:
|
| 26 |
+
return await self._search_protocols(query)
|
| 27 |
+
else:
|
| 28 |
+
return await self._get_top_protocols()
|
| 29 |
+
|
| 30 |
+
except Exception as e:
|
| 31 |
+
return f"DeFiLlama error: {str(e)}"
|
| 32 |
+
|
| 33 |
+
async def _get_top_protocols(self) -> str:
|
| 34 |
+
data = await self.make_request(f"{self.base_url}/protocols")
|
| 35 |
+
|
| 36 |
+
if not data:
|
| 37 |
+
return "No DeFi protocol data available"
|
| 38 |
+
|
| 39 |
+
top_protocols = sorted(data, key=lambda x: x.get("tvl", 0), reverse=True)[:10]
|
| 40 |
+
|
| 41 |
+
result = "π¦ **Top DeFi Protocols by TVL:**\n\n"
|
| 42 |
+
|
| 43 |
+
for i, protocol in enumerate(top_protocols, 1):
|
| 44 |
+
name = protocol.get("name", "Unknown")
|
| 45 |
+
tvl = protocol.get("tvl", 0)
|
| 46 |
+
change = protocol.get("change_1d", 0)
|
| 47 |
+
chain = protocol.get("chain", "Multi-chain")
|
| 48 |
+
|
| 49 |
+
emoji = "π" if change >= 0 else "π"
|
| 50 |
+
result += f"{i}. **{name}** ({chain}): ${tvl/1e9:.2f}B TVL {emoji} ({change:+.2f}%)\n"
|
| 51 |
+
|
| 52 |
+
return result
|
| 53 |
+
|
| 54 |
+
async def _get_tvl_overview(self) -> str:
|
| 55 |
+
try:
|
| 56 |
+
protocols_data = await self.make_request(f"{self.base_url}/protocols")
|
| 57 |
+
chains_data = await self.make_request(f"{self.base_url}/chains")
|
| 58 |
+
|
| 59 |
+
if not protocols_data or not chains_data:
|
| 60 |
+
return "TVL overview data unavailable"
|
| 61 |
+
|
| 62 |
+
total_tvl = sum(p.get("tvl", 0) for p in protocols_data)
|
| 63 |
+
top_chains = sorted(chains_data, key=lambda x: x.get("tvl", 0), reverse=True)[:5]
|
| 64 |
+
|
| 65 |
+
result = "π **DeFi TVL Overview:**\n\n"
|
| 66 |
+
result += f"π° **Total TVL**: ${total_tvl/1e9:.2f}B\n\n"
|
| 67 |
+
result += "**Top Chains by TVL:**\n"
|
| 68 |
+
|
| 69 |
+
for i, chain in enumerate(top_chains, 1):
|
| 70 |
+
name = chain.get("name", "Unknown")
|
| 71 |
+
tvl = chain.get("tvl", 0)
|
| 72 |
+
result += f"{i}. **{name}**: ${tvl/1e9:.2f}B\n"
|
| 73 |
+
|
| 74 |
+
return result
|
| 75 |
+
|
| 76 |
+
except Exception:
|
| 77 |
+
return await self._get_top_protocols()
|
| 78 |
+
|
| 79 |
+
async def _get_protocol_data(self, protocol: str) -> str:
|
| 80 |
+
protocols = await self.make_request(f"{self.base_url}/protocols")
|
| 81 |
+
|
| 82 |
+
if not protocols:
|
| 83 |
+
return f"No data available for {protocol}"
|
| 84 |
+
|
| 85 |
+
matching_protocol = None
|
| 86 |
+
for p in protocols:
|
| 87 |
+
if protocol.lower() in p.get("name", "").lower():
|
| 88 |
+
matching_protocol = p
|
| 89 |
+
break
|
| 90 |
+
|
| 91 |
+
if not matching_protocol:
|
| 92 |
+
return f"Protocol '{protocol}' not found"
|
| 93 |
+
|
| 94 |
+
name = matching_protocol.get("name", "Unknown")
|
| 95 |
+
tvl = matching_protocol.get("tvl", 0)
|
| 96 |
+
change_1d = matching_protocol.get("change_1d", 0)
|
| 97 |
+
change_7d = matching_protocol.get("change_7d", 0)
|
| 98 |
+
chain = matching_protocol.get("chain", "Multi-chain")
|
| 99 |
+
category = matching_protocol.get("category", "Unknown")
|
| 100 |
+
|
| 101 |
+
result = f"ποΈ **{name} Protocol Analysis:**\n\n"
|
| 102 |
+
result += f"π° **TVL**: ${tvl/1e9:.2f}B\n"
|
| 103 |
+
result += f"π **24h Change**: {change_1d:+.2f}%\n"
|
| 104 |
+
result += f"π **7d Change**: {change_7d:+.2f}%\n"
|
| 105 |
+
result += f"βοΈ **Chain**: {chain}\n"
|
| 106 |
+
result += f"π·οΈ **Category**: {category}\n"
|
| 107 |
+
|
| 108 |
+
return result
|
| 109 |
+
|
| 110 |
+
async def _search_protocols(self, query: str) -> str:
|
| 111 |
+
protocols = await self.make_request(f"{self.base_url}/protocols")
|
| 112 |
+
|
| 113 |
+
if not protocols:
|
| 114 |
+
return "No protocol data available"
|
| 115 |
+
|
| 116 |
+
matching = [p for p in protocols if query.lower() in p.get("name", "").lower()][:5]
|
| 117 |
+
|
| 118 |
+
if not matching:
|
| 119 |
+
return f"No protocols found matching '{query}'"
|
| 120 |
+
|
| 121 |
+
result = f"π **Protocols matching '{query}':**\n\n"
|
| 122 |
+
|
| 123 |
+
for protocol in matching:
|
| 124 |
+
name = protocol.get("name", "Unknown")
|
| 125 |
+
tvl = protocol.get("tvl", 0)
|
| 126 |
+
chain = protocol.get("chain", "Multi-chain")
|
| 127 |
+
result += f"β’ **{name}** ({chain}): ${tvl/1e9:.2f}B TVL\n"
|
| 128 |
+
|
| 129 |
+
return result
|
src/tools/etherscan_tool.py
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any, Optional
|
| 2 |
+
from pydantic import BaseModel, PrivateAttr
|
| 3 |
+
from src.tools.base_tool import BaseWeb3Tool, Web3ToolInput
|
| 4 |
+
from src.utils.config import config
|
| 5 |
+
|
| 6 |
+
class EtherscanTool(BaseWeb3Tool):
|
| 7 |
+
name: str = "etherscan_data"
|
| 8 |
+
description: str = """Get Ethereum blockchain data from Etherscan.
|
| 9 |
+
Useful for: transaction analysis, address information, gas prices, token data.
|
| 10 |
+
Input: Ethereum address, transaction hash, or general blockchain query."""
|
| 11 |
+
args_schema: type[BaseModel] = Web3ToolInput
|
| 12 |
+
|
| 13 |
+
_base_url: str = PrivateAttr(default="https://api.etherscan.io/api")
|
| 14 |
+
_api_key: Optional[str] = PrivateAttr(default=None)
|
| 15 |
+
|
| 16 |
+
def __init__(self):
|
| 17 |
+
super().__init__()
|
| 18 |
+
self._api_key = config.ETHERSCAN_API_KEY
|
| 19 |
+
|
| 20 |
+
async def _arun(self, query: str, filters: Optional[Dict[str, Any]] = None) -> str:
|
| 21 |
+
if not self.api_key:
|
| 22 |
+
return "β Etherscan API key not configured"
|
| 23 |
+
|
| 24 |
+
try:
|
| 25 |
+
filters = filters or {}
|
| 26 |
+
|
| 27 |
+
if filters.get("type") == "gas_prices":
|
| 28 |
+
return await self._get_gas_prices()
|
| 29 |
+
elif filters.get("type") == "eth_stats":
|
| 30 |
+
return await self._get_eth_stats()
|
| 31 |
+
elif self._is_address(query):
|
| 32 |
+
return await self._get_address_info(query)
|
| 33 |
+
elif self._is_tx_hash(query):
|
| 34 |
+
return await self._get_transaction_info(query)
|
| 35 |
+
else:
|
| 36 |
+
return await self._get_gas_prices()
|
| 37 |
+
|
| 38 |
+
except Exception as e:
|
| 39 |
+
return f"Etherscan error: {str(e)}"
|
| 40 |
+
|
| 41 |
+
def _is_address(self, query: str) -> bool:
|
| 42 |
+
return len(query) == 42 and query.startswith("0x")
|
| 43 |
+
|
| 44 |
+
def _is_tx_hash(self, query: str) -> bool:
|
| 45 |
+
return len(query) == 66 and query.startswith("0x")
|
| 46 |
+
|
| 47 |
+
async def _get_gas_prices(self) -> str:
|
| 48 |
+
params = {
|
| 49 |
+
"module": "gastracker",
|
| 50 |
+
"action": "gasoracle",
|
| 51 |
+
"apikey": self.api_key
|
| 52 |
+
}
|
| 53 |
+
|
| 54 |
+
data = await self.make_request(self.base_url, params)
|
| 55 |
+
|
| 56 |
+
if data.get("status") != "1":
|
| 57 |
+
return "Gas price data unavailable"
|
| 58 |
+
|
| 59 |
+
result_data = data.get("result", {})
|
| 60 |
+
safe_gas = result_data.get("SafeGasPrice", "N/A")
|
| 61 |
+
standard_gas = result_data.get("StandardGasPrice", "N/A")
|
| 62 |
+
fast_gas = result_data.get("FastGasPrice", "N/A")
|
| 63 |
+
|
| 64 |
+
result = "β½ **Ethereum Gas Prices:**\n\n"
|
| 65 |
+
result += f"π **Safe**: {safe_gas} gwei\n"
|
| 66 |
+
result += f"β‘ **Standard**: {standard_gas} gwei\n"
|
| 67 |
+
result += f"π **Fast**: {fast_gas} gwei\n"
|
| 68 |
+
|
| 69 |
+
return result
|
| 70 |
+
|
| 71 |
+
async def _get_eth_stats(self) -> str:
|
| 72 |
+
params = {
|
| 73 |
+
"module": "stats",
|
| 74 |
+
"action": "ethsupply",
|
| 75 |
+
"apikey": self.api_key
|
| 76 |
+
}
|
| 77 |
+
|
| 78 |
+
data = await self.make_request(self.base_url, params)
|
| 79 |
+
|
| 80 |
+
if data.get("status") != "1":
|
| 81 |
+
return "Ethereum stats unavailable"
|
| 82 |
+
|
| 83 |
+
eth_supply = int(data.get("result", 0)) / 1e18
|
| 84 |
+
|
| 85 |
+
result = "π **Ethereum Network Stats:**\n\n"
|
| 86 |
+
result += f"π **ETH Supply**: {eth_supply:,.0f} ETH\n"
|
| 87 |
+
|
| 88 |
+
return result
|
| 89 |
+
|
| 90 |
+
async def _get_address_info(self, address: str) -> str:
|
| 91 |
+
params = {
|
| 92 |
+
"module": "account",
|
| 93 |
+
"action": "balance",
|
| 94 |
+
"address": address,
|
| 95 |
+
"tag": "latest",
|
| 96 |
+
"apikey": self.api_key
|
| 97 |
+
}
|
| 98 |
+
|
| 99 |
+
data = await self.make_request(self.base_url, params)
|
| 100 |
+
|
| 101 |
+
if data.get("status") != "1":
|
| 102 |
+
return f"Address information unavailable for {address}"
|
| 103 |
+
|
| 104 |
+
balance_wei = int(data.get("result", 0))
|
| 105 |
+
balance_eth = balance_wei / 1e18
|
| 106 |
+
|
| 107 |
+
result = f"π **Address Information:**\n\n"
|
| 108 |
+
result += f"**Address**: {address}\n"
|
| 109 |
+
result += f"π° **Balance**: {balance_eth:.4f} ETH\n"
|
| 110 |
+
|
| 111 |
+
return result
|
| 112 |
+
|
| 113 |
+
async def _get_transaction_info(self, tx_hash: str) -> str:
|
| 114 |
+
params = {
|
| 115 |
+
"module": "proxy",
|
| 116 |
+
"action": "eth_getTransactionByHash",
|
| 117 |
+
"txhash": tx_hash,
|
| 118 |
+
"apikey": self.api_key
|
| 119 |
+
}
|
| 120 |
+
|
| 121 |
+
data = await self.make_request(self.base_url, params)
|
| 122 |
+
|
| 123 |
+
if not data.get("result"):
|
| 124 |
+
return f"Transaction not found: {tx_hash}"
|
| 125 |
+
|
| 126 |
+
tx = data.get("result", {})
|
| 127 |
+
value_wei = int(tx.get("value", "0x0"), 16)
|
| 128 |
+
value_eth = value_wei / 1e18
|
| 129 |
+
gas_price = int(tx.get("gasPrice", "0x0"), 16) / 1e9
|
| 130 |
+
|
| 131 |
+
result = f"π **Transaction Information:**\n\n"
|
| 132 |
+
result += f"**Hash**: {tx_hash}\n"
|
| 133 |
+
result += f"**From**: {tx.get('from', 'N/A')}\n"
|
| 134 |
+
result += f"**To**: {tx.get('to', 'N/A')}\n"
|
| 135 |
+
result += f"π° **Value**: {value_eth:.4f} ETH\n"
|
| 136 |
+
result += f"β½ **Gas Price**: {gas_price:.2f} gwei\n"
|
| 137 |
+
|
| 138 |
+
return result
|
src/utils/__init__.py
ADDED
|
File without changes
|
src/utils/config.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
from src.config import config
|
src/utils/exceptions.py
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from typing import Dict, Any
|
| 2 |
+
|
| 3 |
+
class CustomException(Exception):
|
| 4 |
+
pass
|
| 5 |
+
|
| 6 |
+
class APIError(CustomException):
|
| 7 |
+
pass
|
| 8 |
+
|
| 9 |
+
class RateLimitError(APIError):
|
| 10 |
+
pass
|
| 11 |
+
|
| 12 |
+
class DataValidationError(CustomException):
|
| 13 |
+
pass
|
| 14 |
+
|
| 15 |
+
class ConfigurationError(CustomException):
|
| 16 |
+
pass
|
src/utils/logger.py
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import logging
|
| 2 |
+
import sys
|
| 3 |
+
from typing import Optional
|
| 4 |
+
|
| 5 |
+
def get_logger(name: str, level: str = "INFO") -> logging.Logger:
|
| 6 |
+
logger = logging.getLogger(name)
|
| 7 |
+
|
| 8 |
+
if not logger.handlers:
|
| 9 |
+
handler = logging.StreamHandler(sys.stdout)
|
| 10 |
+
formatter = logging.Formatter(
|
| 11 |
+
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
| 12 |
+
)
|
| 13 |
+
handler.setFormatter(formatter)
|
| 14 |
+
logger.addHandler(handler)
|
| 15 |
+
logger.setLevel(getattr(logging, level.upper()))
|
| 16 |
+
|
| 17 |
+
return logger
|
src/visualizations.py
CHANGED
|
@@ -1,238 +1,62 @@
|
|
| 1 |
import plotly.graph_objects as go
|
| 2 |
-
import plotly.express as px
|
| 3 |
-
from plotly.subplots import make_subplots
|
| 4 |
-
from typing import Dict, Any, List
|
| 5 |
-
import pandas as pd
|
| 6 |
from datetime import datetime
|
|
|
|
| 7 |
|
| 8 |
-
def create_price_chart(data: Dict[str, Any], symbol: str) ->
|
| 9 |
try:
|
| 10 |
if not data or "prices" not in data:
|
| 11 |
-
return f"
|
| 12 |
|
| 13 |
prices = data["prices"]
|
| 14 |
-
|
|
|
|
| 15 |
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
fig = make_subplots(
|
| 23 |
-
rows=2, cols=1,
|
| 24 |
-
shared_xaxes=True,
|
| 25 |
-
vertical_spacing=0.05,
|
| 26 |
-
subplot_titles=[f"{symbol.upper()} Price Chart", "Volume"],
|
| 27 |
-
row_width=[0.7, 0.3]
|
| 28 |
-
)
|
| 29 |
-
|
| 30 |
-
fig.add_trace(
|
| 31 |
-
go.Scatter(
|
| 32 |
-
x=df["datetime"],
|
| 33 |
-
y=df["price"],
|
| 34 |
-
mode="lines",
|
| 35 |
-
name="Price",
|
| 36 |
-
line=dict(color="#00D4AA", width=2),
|
| 37 |
-
hovertemplate="<b>%{y:$,.2f}</b><br>%{x}<extra></extra>"
|
| 38 |
-
),
|
| 39 |
-
row=1, col=1
|
| 40 |
-
)
|
| 41 |
-
|
| 42 |
-
if volumes:
|
| 43 |
-
vol_df = pd.DataFrame(volumes, columns=["timestamp", "volume"])
|
| 44 |
-
vol_df["datetime"] = pd.to_datetime(vol_df["timestamp"], unit="ms")
|
| 45 |
-
|
| 46 |
-
fig.add_trace(
|
| 47 |
-
go.Bar(
|
| 48 |
-
x=vol_df["datetime"],
|
| 49 |
-
y=vol_df["volume"],
|
| 50 |
-
name="Volume",
|
| 51 |
-
marker_color="#FF6B6B",
|
| 52 |
-
opacity=0.7,
|
| 53 |
-
hovertemplate="<b>$%{y:,.0f}</b><br>%{x}<extra></extra>"
|
| 54 |
-
),
|
| 55 |
-
row=2, col=1
|
| 56 |
-
)
|
| 57 |
-
|
| 58 |
-
current_price = df["price"].iloc[-1]
|
| 59 |
-
price_change = ((df["price"].iloc[-1] - df["price"].iloc[0]) / df["price"].iloc[0]) * 100
|
| 60 |
|
| 61 |
fig.update_layout(
|
| 62 |
-
title=
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
font=dict(size=20, color="#FFFFFF")
|
| 66 |
-
),
|
| 67 |
-
xaxis_title="Date",
|
| 68 |
-
yaxis_title="Price (USD)",
|
| 69 |
-
template="plotly_dark",
|
| 70 |
-
showlegend=False,
|
| 71 |
-
height=600,
|
| 72 |
-
margin=dict(l=60, r=60, t=80, b=60),
|
| 73 |
-
plot_bgcolor="rgba(0,0,0,0)",
|
| 74 |
-
paper_bgcolor="rgba(0,0,0,0)"
|
| 75 |
)
|
| 76 |
|
| 77 |
-
fig
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
return fig.to_html(include_plotlyjs="cdn", div_id=f"chart_{symbol}")
|
| 81 |
-
|
| 82 |
-
except Exception as e:
|
| 83 |
-
return f"<div style='padding: 20px; text-align: center; color: #FF6B6B;'>Chart generation failed: {str(e)}</div>"
|
| 84 |
|
| 85 |
-
def create_market_overview(data: Dict[str, Any]) ->
|
| 86 |
try:
|
| 87 |
-
if not data
|
| 88 |
-
return "
|
| 89 |
-
|
| 90 |
-
market_data = data["market_data"]
|
| 91 |
-
if not market_data:
|
| 92 |
-
return "<div style='padding: 20px; text-align: center;'>No market data found</div>"
|
| 93 |
-
|
| 94 |
-
df = pd.DataFrame(market_data)
|
| 95 |
-
df = df.head(20)
|
| 96 |
-
|
| 97 |
-
fig = make_subplots(
|
| 98 |
-
rows=2, cols=2,
|
| 99 |
-
subplot_titles=[
|
| 100 |
-
"Market Cap Distribution",
|
| 101 |
-
"24h Price Changes",
|
| 102 |
-
"Trading Volume",
|
| 103 |
-
"Price vs Volume"
|
| 104 |
-
],
|
| 105 |
-
specs=[[{"type": "pie"}, {"type": "bar"}],
|
| 106 |
-
[{"type": "bar"}, {"type": "scatter"}]]
|
| 107 |
-
)
|
| 108 |
-
|
| 109 |
-
fig.add_trace(
|
| 110 |
-
go.Pie(
|
| 111 |
-
labels=df["symbol"].str.upper(),
|
| 112 |
-
values=df["market_cap"],
|
| 113 |
-
textinfo="label+percent",
|
| 114 |
-
textposition="inside",
|
| 115 |
-
marker=dict(colors=px.colors.qualitative.Set3),
|
| 116 |
-
hovertemplate="<b>%{label}</b><br>Market Cap: $%{value:,.0f}<extra></extra>"
|
| 117 |
-
),
|
| 118 |
-
row=1, col=1
|
| 119 |
-
)
|
| 120 |
-
|
| 121 |
-
colors = ["#00D4AA" if x >= 0 else "#FF6B6B" for x in df["price_change_percentage_24h"]]
|
| 122 |
-
fig.add_trace(
|
| 123 |
-
go.Bar(
|
| 124 |
-
x=df["symbol"].str.upper(),
|
| 125 |
-
y=df["price_change_percentage_24h"],
|
| 126 |
-
marker_color=colors,
|
| 127 |
-
hovertemplate="<b>%{x}</b><br>24h Change: %{y:+.2f}%<extra></extra>"
|
| 128 |
-
),
|
| 129 |
-
row=1, col=2
|
| 130 |
-
)
|
| 131 |
|
| 132 |
-
fig.
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
hovertemplate="<b>%{x}</b><br>Volume: $%{y:,.0f}<extra></extra>"
|
| 138 |
-
),
|
| 139 |
-
row=2, col=1
|
| 140 |
)
|
| 141 |
|
| 142 |
-
fig.add_trace(
|
| 143 |
-
go.Scatter(
|
| 144 |
-
x=df["current_price"],
|
| 145 |
-
y=df["total_volume"],
|
| 146 |
-
mode="markers+text",
|
| 147 |
-
text=df["symbol"].str.upper(),
|
| 148 |
-
textposition="top center",
|
| 149 |
-
marker=dict(
|
| 150 |
-
size=df["market_cap"] / df["market_cap"].max() * 50 + 10,
|
| 151 |
-
color=df["price_change_percentage_24h"],
|
| 152 |
-
colorscale="RdYlGn",
|
| 153 |
-
colorbar=dict(title="24h Change %"),
|
| 154 |
-
line=dict(width=1, color="white")
|
| 155 |
-
),
|
| 156 |
-
hovertemplate="<b>%{text}</b><br>Price: $%{x:,.4f}<br>Volume: $%{y:,.0f}<extra></extra>"
|
| 157 |
-
),
|
| 158 |
-
row=2, col=2
|
| 159 |
-
)
|
| 160 |
-
|
| 161 |
-
global_info = data.get("global_data", {}).get("data", {})
|
| 162 |
-
total_mcap = global_info.get("total_market_cap", {}).get("usd", 0)
|
| 163 |
-
total_volume = global_info.get("total_volume", {}).get("usd", 0)
|
| 164 |
-
btc_dominance = global_info.get("market_cap_percentage", {}).get("btc", 0)
|
| 165 |
-
|
| 166 |
-
title_text = f"Crypto Market Overview - Total MCap: ${total_mcap/1e12:.2f}T | 24h Vol: ${total_volume/1e9:.0f}B | BTC Dom: {btc_dominance:.1f}%"
|
| 167 |
-
|
| 168 |
fig.update_layout(
|
| 169 |
-
title=
|
| 170 |
-
|
| 171 |
-
x=0.5,
|
| 172 |
-
font=dict(size=16, color="#FFFFFF")
|
| 173 |
-
),
|
| 174 |
-
template="plotly_dark",
|
| 175 |
-
showlegend=False,
|
| 176 |
-
height=800,
|
| 177 |
-
margin=dict(l=60, r=60, t=100, b=60),
|
| 178 |
-
plot_bgcolor="rgba(0,0,0,0)",
|
| 179 |
-
paper_bgcolor="rgba(0,0,0,0)"
|
| 180 |
)
|
| 181 |
|
| 182 |
-
fig
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
return fig.to_html(include_plotlyjs="cdn", div_id="market_overview")
|
| 186 |
-
|
| 187 |
-
except Exception as e:
|
| 188 |
-
return f"<div style='padding: 20px; text-align: center; color: #FF6B6B;'>Market overview failed: {str(e)}</div>"
|
| 189 |
|
| 190 |
-
def
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
| 196 |
-
|
| 197 |
-
|
| 198 |
-
|
| 199 |
-
|
| 200 |
-
|
| 201 |
-
|
| 202 |
-
colors = px.colors.qualitative.Set1[:len(df)]
|
| 203 |
-
|
| 204 |
-
for i, (_, coin) in enumerate(df.iterrows()):
|
| 205 |
-
fig.add_trace(
|
| 206 |
-
go.Bar(
|
| 207 |
-
name=coin["symbol"].upper(),
|
| 208 |
-
x=[coin["symbol"].upper()],
|
| 209 |
-
y=[coin["current_price"]],
|
| 210 |
-
marker_color=colors[i],
|
| 211 |
-
hovertemplate=f"<b>{coin['name']}</b><br>Price: $%{{y:,.4f}}<extra></extra>"
|
| 212 |
-
),
|
| 213 |
-
row=1, col=1
|
| 214 |
-
)
|
| 215 |
-
|
| 216 |
-
fig.add_trace(
|
| 217 |
-
go.Bar(
|
| 218 |
-
name=coin["symbol"].upper(),
|
| 219 |
-
x=[coin["symbol"].upper()],
|
| 220 |
-
y=[coin["market_cap"]],
|
| 221 |
-
marker_color=colors[i],
|
| 222 |
-
showlegend=False,
|
| 223 |
-
hovertemplate=f"<b>{coin['name']}</b><br>Market Cap: $%{{y:,.0f}}<extra></extra>"
|
| 224 |
-
),
|
| 225 |
-
row=1, col=2
|
| 226 |
-
)
|
| 227 |
-
|
| 228 |
-
fig.update_layout(
|
| 229 |
-
title="Cryptocurrency Comparison",
|
| 230 |
-
template="plotly_dark",
|
| 231 |
-
height=500,
|
| 232 |
-
showlegend=True
|
| 233 |
-
)
|
| 234 |
-
|
| 235 |
-
return fig.to_html(include_plotlyjs="cdn", div_id="comparison_chart")
|
| 236 |
-
|
| 237 |
-
except Exception as e:
|
| 238 |
-
return f"<div style='padding: 20px; text-align: center; color: #FF6B6B;'>Comparison chart failed: {str(e)}</div>"
|
|
|
|
| 1 |
import plotly.graph_objects as go
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
from datetime import datetime
|
| 3 |
+
from typing import Dict, Any
|
| 4 |
|
| 5 |
+
def create_price_chart(data: Dict[str, Any], symbol: str) -> go.Figure:
|
| 6 |
try:
|
| 7 |
if not data or "prices" not in data:
|
| 8 |
+
return _empty_chart(f"No price data for {symbol}")
|
| 9 |
|
| 10 |
prices = data["prices"]
|
| 11 |
+
timestamps = [datetime.fromtimestamp(p[0]/1000) for p in prices]
|
| 12 |
+
values = [p[1] for p in prices]
|
| 13 |
|
| 14 |
+
fig = go.Figure()
|
| 15 |
+
fig.add_trace(go.Scatter(
|
| 16 |
+
x=timestamps, y=values, mode='lines',
|
| 17 |
+
name=f'{symbol.upper()} Price',
|
| 18 |
+
line=dict(color='#00D4AA', width=2)
|
| 19 |
+
))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
fig.update_layout(
|
| 22 |
+
title=f'{symbol.upper()} Price History',
|
| 23 |
+
xaxis_title='Date', yaxis_title='Price (USD)',
|
| 24 |
+
template='plotly_dark', height=400
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
)
|
| 26 |
|
| 27 |
+
return fig
|
| 28 |
+
except Exception:
|
| 29 |
+
return _empty_chart(f"Chart error for {symbol}")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
+
def create_market_overview(data: Dict[str, Any]) -> go.Figure:
|
| 32 |
try:
|
| 33 |
+
if not data:
|
| 34 |
+
return _empty_chart("No market data available")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
+
fig = go.Figure()
|
| 37 |
+
fig.add_annotation(
|
| 38 |
+
text="Market Overview\n" + str(data)[:200] + "...",
|
| 39 |
+
x=0.5, y=0.5, font=dict(size=12, color="white"),
|
| 40 |
+
showarrow=False, align="left"
|
|
|
|
|
|
|
|
|
|
| 41 |
)
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
fig.update_layout(
|
| 44 |
+
title="Market Overview", template='plotly_dark', height=400,
|
| 45 |
+
xaxis=dict(visible=False), yaxis=dict(visible=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
)
|
| 47 |
|
| 48 |
+
return fig
|
| 49 |
+
except Exception:
|
| 50 |
+
return _empty_chart("Market overview error")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
+
def _empty_chart(message: str) -> go.Figure:
|
| 53 |
+
fig = go.Figure()
|
| 54 |
+
fig.add_annotation(
|
| 55 |
+
text=message, x=0.5, y=0.5,
|
| 56 |
+
font=dict(size=16, color="white"), showarrow=False
|
| 57 |
+
)
|
| 58 |
+
fig.update_layout(
|
| 59 |
+
template='plotly_dark', height=400,
|
| 60 |
+
xaxis=dict(visible=False), yaxis=dict(visible=False)
|
| 61 |
+
)
|
| 62 |
+
return fig
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
test_app.py
ADDED
|
@@ -0,0 +1,108 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
|
| 3 |
+
import sys
|
| 4 |
+
import os
|
| 5 |
+
|
| 6 |
+
sys.path.insert(0, os.path.dirname(__file__))
|
| 7 |
+
|
| 8 |
+
def test_imports():
|
| 9 |
+
try:
|
| 10 |
+
print("Testing imports...")
|
| 11 |
+
|
| 12 |
+
# Test basic Python modules
|
| 13 |
+
import json
|
| 14 |
+
import asyncio
|
| 15 |
+
from datetime import datetime
|
| 16 |
+
from typing import List, Tuple
|
| 17 |
+
print("β
Basic Python modules imported successfully")
|
| 18 |
+
|
| 19 |
+
# Test installed packages
|
| 20 |
+
import gradio as gr
|
| 21 |
+
print("β
Gradio imported successfully")
|
| 22 |
+
|
| 23 |
+
import aiohttp
|
| 24 |
+
print("β
aiohttp imported successfully")
|
| 25 |
+
|
| 26 |
+
import plotly
|
| 27 |
+
print("β
Plotly imported successfully")
|
| 28 |
+
|
| 29 |
+
import pandas
|
| 30 |
+
print("β
Pandas imported successfully")
|
| 31 |
+
|
| 32 |
+
import pydantic
|
| 33 |
+
print("β
Pydantic imported successfully")
|
| 34 |
+
|
| 35 |
+
# Test LangChain
|
| 36 |
+
import langchain
|
| 37 |
+
from langchain.agents import AgentExecutor
|
| 38 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 39 |
+
print("β
LangChain modules imported successfully")
|
| 40 |
+
|
| 41 |
+
# Test our modules
|
| 42 |
+
from src.utils.config import config
|
| 43 |
+
from src.utils.logger import get_logger
|
| 44 |
+
print("β
Config and logger imported successfully")
|
| 45 |
+
|
| 46 |
+
from src.tools.base_tool import BaseWeb3Tool
|
| 47 |
+
from src.tools.coingecko_tool import CoinGeckoTool
|
| 48 |
+
from src.tools.defillama_tool import DeFiLlamaTool
|
| 49 |
+
from src.tools.etherscan_tool import EtherscanTool
|
| 50 |
+
print("β
Tools imported successfully")
|
| 51 |
+
|
| 52 |
+
from src.agent.research_agent import Web3ResearchAgent
|
| 53 |
+
from src.agent.query_planner import QueryPlanner
|
| 54 |
+
print("β
Agent modules imported successfully")
|
| 55 |
+
|
| 56 |
+
from src.api.airaa_integration import AIRAAIntegration
|
| 57 |
+
print("β
AIRAA integration imported successfully")
|
| 58 |
+
|
| 59 |
+
from src.visualizations import create_price_chart, create_market_overview
|
| 60 |
+
print("β
Visualizations imported successfully")
|
| 61 |
+
|
| 62 |
+
# Test app import
|
| 63 |
+
from app import Web3CoPilotApp
|
| 64 |
+
print("β
Main app imported successfully")
|
| 65 |
+
|
| 66 |
+
print("\nπ All imports successful! The application is ready to run.")
|
| 67 |
+
return True
|
| 68 |
+
|
| 69 |
+
except ImportError as e:
|
| 70 |
+
print(f"β Import error: {e}")
|
| 71 |
+
return False
|
| 72 |
+
except Exception as e:
|
| 73 |
+
print(f"β Unexpected error: {e}")
|
| 74 |
+
return False
|
| 75 |
+
|
| 76 |
+
def test_app_initialization():
|
| 77 |
+
try:
|
| 78 |
+
print("\nTesting app initialization...")
|
| 79 |
+
# This will test if we can create the app instance
|
| 80 |
+
# but won't actually run it
|
| 81 |
+
os.environ.setdefault('GEMINI_API_KEY', 'test_key_for_import_test')
|
| 82 |
+
|
| 83 |
+
from app import Web3CoPilotApp
|
| 84 |
+
print("β
App class imported successfully")
|
| 85 |
+
|
| 86 |
+
# Test if we can create the interface (but don't launch)
|
| 87 |
+
app = Web3CoPilotApp()
|
| 88 |
+
interface = app.create_interface()
|
| 89 |
+
print("β
App interface created successfully")
|
| 90 |
+
|
| 91 |
+
print("\nπ Application is fully functional and ready to launch!")
|
| 92 |
+
return True
|
| 93 |
+
|
| 94 |
+
except Exception as e:
|
| 95 |
+
print(f"β App initialization error: {e}")
|
| 96 |
+
return False
|
| 97 |
+
|
| 98 |
+
if __name__ == "__main__":
|
| 99 |
+
print("=" * 60)
|
| 100 |
+
print("Web3 Research Co-Pilot - Application Test")
|
| 101 |
+
print("=" * 60)
|
| 102 |
+
|
| 103 |
+
success = test_imports()
|
| 104 |
+
if success:
|
| 105 |
+
test_app_initialization()
|
| 106 |
+
|
| 107 |
+
print("=" * 60)
|
| 108 |
+
print("Test complete!")
|