Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -1,27 +1,27 @@
 from crewai import Agent
-from langchain.llms import HuggingFaceHub
-from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain_community.llms import HuggingFaceHub
+from langchain.llms import HuggingFacePipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from crypto_tools import CryptoTools
 from news_tools import NewsTools
 from sentiment_tools import SentimentTools
-import os
-
 
 class CryptoAnalysisAgents:
     def __init__(self):
-        self.llm = HuggingFaceHub(
-            repo_id="facebook/opt-125m", 
-            model_kwargs={"temperature": 0.7, "max_length": 512},
-            task="text-generation"
-        )
-        self.embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
-        huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+        # Load Phi-2 model and tokenizer
+        model_name = "microsoft/phi-2"
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForCausalLM.from_pretrained(model_name)
+        
+        # Create a text-generation pipeline
+        pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512)
         
+        # Create a LangChain wrapper around the Hugging Face pipeline
+        self.llm = HuggingFacePipeline(pipeline=pipe)
+
     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",
+            goal="Provide in-depth industry, market analysis and insights for cryptocurrencies. You also provide insight 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,
             llm=self.llm,
@@ -60,4 +60,4 @@ class CryptoAnalysisAgents:
                 NewsTools.search_crypto_news,
                 SentimentTools.analyze_sentiment
             ]
-        )
+        )