Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -1,22 +1,24 @@
+# crypto_analysis_agents.py - CrewAI Native Version
 from crewai import Agent
-from langchain_together import ChatTogether, Together # <-- Import Together class here
-from langchain_huggingface import HuggingFacePipeline
+from langchain_together import ChatTogether, Together
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, GenerationConfig
-from crypto_tools import get_crypto_price, get_market_cap, calculate_rsi, calculate_moving_average
-from news_tools import search_crypto_news
-from sentiment_tools import analyze_sentiment
 import os
 from dotenv import load_dotenv
 
-load_dotenv()
+# Import the CrewAI native tool classes
+from crypto_tools import CryptoPriceTool, MarketCapTool, RSITool, MovingAverageTool
+from news_tools import NewsSearchTool
+from sentiment_tools import SentimentTool
+
 
+load_dotenv()
 TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
 
 class CryptoAnalysisAgents:
     def __init__(self):
-        # Use Together AI's fast inference instead of local models
+        # Use Together AI's fast inference
         self.llm = Together(
-            model="meta-llama/Llama-2-7b-chat-hf",  # Fast, efficient model
+            model="meta-llama/Llama-2-7b-chat-hf",
             together_api_key=TOGETHER_API_KEY,
             temperature=0.7,
             max_tokens=512,
@@ -28,12 +30,12 @@ class CryptoAnalysisAgents:
             role='Crypto Market Analyst',
             goal="Provide concise market analysis and insights for cryptocurrencies",
             backstory="Expert crypto analyst with focus on market trends and news impact",
-            verbose=False,  # Reduce verbosity for speed
+            verbose=False,
             llm=self.llm,
             tools=[
-                get_crypto_price,
-                get_market_cap,
-                search_crypto_news
+                CryptoPriceTool(),
+                MarketCapTool(),
+                NewsSearchTool()
             ]
         )
 
@@ -45,9 +47,9 @@ class CryptoAnalysisAgents:
             verbose=False,
             llm=self.llm,
             tools=[
-                get_crypto_price,
-                calculate_rsi,
-                calculate_moving_average
+                CryptoPriceTool(),
+                RSITool(),
+                MovingAverageTool()
             ]
         )
 
@@ -59,7 +61,7 @@ class CryptoAnalysisAgents:
             verbose=False,
             llm=self.llm,
             tools=[
-                get_crypto_price,
-                analyze_sentiment
+                CryptoPriceTool(),
+                SentimentTool()
             ]
         )