fix2
Browse files
agent.py
CHANGED
|
@@ -13,7 +13,7 @@ load_dotenv()
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
-
repo_id="
|
| 17 |
task="text-generation",
|
| 18 |
temperature=0.0,
|
| 19 |
huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
|
|
@@ -48,7 +48,6 @@ def web_search(keywords: str, max_results:int = 5) -> str:
|
|
| 48 |
|
| 49 |
class AgentState(TypedDict):
|
| 50 |
messages: List[HumanMessage | AIMessage]
|
| 51 |
-
search_required:
|
| 52 |
|
| 53 |
|
| 54 |
def read_message(state: AgentState) -> AgentState:
|
|
|
|
| 13 |
|
| 14 |
# Base Hugging Face LLM used by the chat wrapper
|
| 15 |
base_llm = HuggingFaceEndpoint(
|
| 16 |
+
repo_id="openai/gpt-oss-20b",
|
| 17 |
task="text-generation",
|
| 18 |
temperature=0.0,
|
| 19 |
huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN"),
|
|
|
|
| 48 |
|
| 49 |
class AgentState(TypedDict):
|
| 50 |
messages: List[HumanMessage | AIMessage]
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
def read_message(state: AgentState) -> AgentState:
|