Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 24 |
-
load_dotenv(".
|
| 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.
|
| 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,
|