Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -2,9 +2,9 @@ from crewai import Agent
 from langchain_together import ChatTogether, Together # <-- Import Together class here
 from langchain_huggingface import HuggingFacePipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, GenerationConfig
-from crypto_tools import CryptoTools
-from news_tools import NewsTools
-from sentiment_tools import SentimentTools
+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
 
@@ -17,16 +17,11 @@ class CryptoAnalysisAgents:
         # Use Together AI's fast inference instead of local models
         self.llm = Together(
             model="meta-llama/Llama-2-7b-chat-hf",  # Fast, efficient model
-            together_api_key=TOGETHER_API_KEY, # Use the loaded variable
+            together_api_key=TOGETHER_API_KEY,
             temperature=0.7,
             max_tokens=512,
             top_p=0.9
         )
-        
-        # Initialize tools once to avoid repeated loading
-        self.crypto_tools = CryptoTools()
-        self.news_tools = NewsTools()
-        self.sentiment_tools = SentimentTools()
 
     def market_analyst(self):
         return Agent(
@@ -36,9 +31,9 @@ class CryptoAnalysisAgents:
             verbose=False,  # Reduce verbosity for speed
             llm=self.llm,
             tools=[
-                self.crypto_tools.get_crypto_price,
-                self.crypto_tools.get_market_cap,
-                self.news_tools.search_crypto_news
+                get_crypto_price,
+                get_market_cap,
+                search_crypto_news
             ]
         )
 
@@ -50,9 +45,9 @@ class CryptoAnalysisAgents:
             verbose=False,
             llm=self.llm,
             tools=[
-                self.crypto_tools.get_crypto_price,
-                self.crypto_tools.calculate_rsi,
-                self.crypto_tools.calculate_moving_average
+                get_crypto_price,
+                calculate_rsi,
+                calculate_moving_average
             ]
         )
 
@@ -64,7 +59,7 @@ class CryptoAnalysisAgents:
             verbose=False,
             llm=self.llm,
             tools=[
-                self.crypto_tools.get_crypto_price,
-                self.sentiment_tools.analyze_sentiment
+                get_crypto_price,
+                analyze_sentiment
             ]
         )