Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -1,31 +1,39 @@
-# crypto_analysis_agents.py - CrewAI Native Version
+# crypto_analysis_agents.py - Refactored to use litellm with Together AI
+
 from crewai import Agent
-from langchain_together import ChatTogether, Together
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, GenerationConfig
-import os
 from dotenv import load_dotenv
+import os
 from litellm import completion
 
-# Import the CrewAI native tool classes
+# Import native tools
 from crypto_tools import CryptoPriceTool, MarketCapTool, RSITool, MovingAverageTool
 from news_tools import NewsSearchTool
 from sentiment_tools import SentimentTool
 
-
+# Load environment variables
 load_dotenv()
-TOGETHER_API_KEY = os.getenv("TOGETHER_API_KEY")
+os.environ["TOGETHERAI_API_KEY"] = os.getenv("TOGETHERAI_API_KEY")  # Ensure environment is set
 
-class CryptoAnalysisAgents:
-    def __init__(self):
-        # Use Together AI's fast inference
-        self.llm = Together(
-            model="togethercomputer/llama-2-7b-chat",
-            together_api_key=TOGETHER_API_KEY,
-            temperature=0.7,
-            max_tokens=512,
-            top_p=0.9
+# Define LLM wrapper using litellm
+class LLMWrapper:
+    def __init__(self, model="together_ai/togethercomputer/llama-2-7b-chat"):
+        self.model = model
+
+    def __call__(self, prompt):
+        messages = [
+            {"role": "user", "content": prompt}
+        ]
+        response = completion(
+            model=self.model,
+            messages=messages
         )
+        return response['choices'][0]['message']['content']
 
+llm = LLMWrapper()
+
+class CryptoAnalysisAgents:
+    def __init__(self):
+        self.llm = llm
 
     def market_analyst(self):
         return Agent(
@@ -66,4 +74,4 @@ class CryptoAnalysisAgents:
                 CryptoPriceTool(),
                 SentimentTool()
             ]
-        )
+        )