jdesiree commited on
Commit
99e37e9
·
verified ·
1 Parent(s): 2374e70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,8 +20,8 @@ from typing import Optional, List, Any, Type
20
  from pydantic import BaseModel, Field
21
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
22
  import torch
23
- # Load environment variables from .EVN fil (case-sensitive)
24
- load_dotenv(".evn")
25
  HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
26
  print("Environment variables loaded.")
27
 
@@ -394,7 +394,7 @@ def create_langchain_agent():
394
  agent = initialize_agent(
395
  tools=tools,
396
  llm=llm,
397
- agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
398
  memory=memory,
399
  verbose=False,
400
  max_iterations=3,
 
20
  from pydantic import BaseModel, Field
21
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
22
  import torch
23
+
24
+ load_dotenv(".env")
25
  HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN")
26
  print("Environment variables loaded.")
27
 
 
394
  agent = initialize_agent(
395
  tools=tools,
396
  llm=llm,
397
+ agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
398
  memory=memory,
399
  verbose=False,
400
  max_iterations=3,