Final_Assignment_Agents_Course

Sleeping

rsobieski commited on Jun 7, 2025

Commit

d8ddea0

verified ·

1 Parent(s): a5257b1

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -16,8 +16,8 @@ qa_dict = {
 def build_graph():
     # Initialize Mistral model
     llm = HuggingFaceEndpoint(
-        # repo_id="mistralai/Mistral-7B-Instruct-v0.3",
-        task="conversation",
         huggingfacehub_api_token=os.environ["HF_TOKEN"]
     )

 def build_graph():
     # Initialize Mistral model
     llm = HuggingFaceEndpoint(
+        repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+        # task="conversation",
         huggingfacehub_api_token=os.environ["HF_TOKEN"]
     )