Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -1,66 +1,57 @@
-import asyncio
-from crewai import Crew
-from textwrap import dedent
-import json
-from crypto_analysis_agents import CryptoAnalysisAgents
-from crypto__analysis_tasks import CryptoAnalysisTasks
-
-class CryptoCrew:
-    def __init__(self, crypto):
-        self.crypto = crypto
-
-    async def run(self):
-        agents = CryptoAnalysisAgents()
-        tasks = CryptoAnalysisTasks()
-
-        market_analyst_agent = agents.market_analyst()
-        technical_analyst_agent = agents.technical_analyst()
-        crypto_advisor_agent = agents.crypto_advisor()
-
-        market_research_task = tasks.market_research(market_analyst_agent, self.crypto)
-        technical_analysis_task = tasks.technical_analysis(technical_analyst_agent)
-        sentiment_analysis_task = tasks.sentiment_analysis(market_analyst_agent)
-        recommend_task = tasks.recommend(crypto_advisor_agent)
-
-        crew = Crew(
-            agents=[
-                market_analyst_agent,
-                technical_analyst_agent,
-                crypto_advisor_agent
-            ],
-            tasks=[
-                market_research_task,
-                technical_analysis_task,
-                sentiment_analysis_task,
-                recommend_task
-            ],
+from crewai import Agent
+from langchain.llms import HuggingFaceHub
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.llms import HuggingFaceHub
+from tools.crypto_tools import CryptoTools
+from tools.news_tools import NewsTools
+from tools.sentiment_tools import SentimentTools
+
+class CryptoAnalysisAgents:
+    def __init__(self):
+        self.llm = HuggingFaceHub(repo_id="bigscience/bloom", model_kwargs={"temperature": 0.7, "max_length": 1024})
+        self.embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+        huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"
+
+    def market_analyst(self):
+        return Agent(
+            role='Crypto Market Analyst',
+            goal="Provide in-depth industry, market analysis and insights for cryptocurrencies. You also provide insight on on our geo-politics and economic policies impact on crypto",
+            backstory="""You are a seasoned crypto market analyst with years of experience in blockchain technology and cryptocurrency markets. Your expertise helps clients navigate the volatile crypto landscape.""",
             verbose=True,
-            iteration_limit=300,
-            time_limit=600
+            llm=self.llm,
+            tools=[
+                CryptoTools.get_crypto_price,
+                CryptoTools.get_market_cap,
+                NewsTools.search_crypto_news,
+                SentimentTools.analyze_sentiment
+            ]
         )
 
-        try:
-            result = await asyncio.wait_for(crew.kickoff(), timeout=600)
-        except asyncio.TimeoutError:
-            return {"summary": "Analysis timed out. Please try again with a simpler query or increased time limit."}
-
-        parsed_result = self.parse_result(result)
-        return parsed_result
-
-    def parse_result(self, result):
-        # Implement your parse_result method here
-        # This is a placeholder implementation
-        return {"summary": str(result)}
+    def technical_analyst(self):
+        return Agent(
+            role='Crypto Technical Analyst',
+            goal="Analyze cryptocurrency price patterns and provide technical insights",
+            backstory="""You are an expert in technical analysis, specializing in cryptocurrency markets. Your chart reading skills and understanding of technical indicators are unparalleled.""",
+            verbose=True,
+            llm=self.llm,
+            tools=[
+                CryptoTools.get_crypto_price,
+                CryptoTools.calculate_rsi,
+                CryptoTools.calculate_moving_average
+            ]
+        )
 
-if __name__ == "__main__":
-    print("## Welcome to Crypto Analysis Crew")
-    print('-------------------------------')
-    crypto = input(dedent("""
-    What is the cryptocurrency you want to analyze?
-    """))
-    crypto_crew = CryptoCrew(crypto)
-    result = asyncio.run(crypto_crew.run())
-    print("\n\n########################")
-    print("## Here is the Report")
-    print("########################\n")
-    print(json.dumps(result, indent=2))
+    def crypto_advisor(self):
+        return Agent(
+            role='Cryptocurrency Investment Advisor',
+            goal="Provide comprehensive investment advice for cryptocurrency portfolios",
+            backstory="""You are a trusted cryptocurrency investment advisor with a deep understanding of blockchain technology, market dynamics, and risk management in the crypto space.""",
+            verbose=True,
+            llm=self.llm,
+            tools=[
+                CryptoTools.get_crypto_price,
+                CryptoTools.get_market_cap,
+                NewsTools.search_crypto_news,
+                SentimentTools.analyze_sentiment
+            ]
+        )