Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -158,7 +158,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
|
|
| 158 |
"""
|
| 159 |
|
| 160 |
agent_config = AgentConfig(
|
| 161 |
-
agent_type = os.getenv("VECTARA_AGENTIC_AGENT_TYPE", AgentType.
|
| 162 |
main_llm_provider = os.getenv("VECTARA_AGENTIC_MAIN_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
| 163 |
main_llm_model_name = os.getenv("VECTARA_AGENTIC_MAIN_MODEL_NAME", ""),
|
| 164 |
tool_llm_provider = os.getenv("VECTARA_AGENTIC_TOOL_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
|
@@ -166,7 +166,7 @@ def initialize_agent(_cfg, agent_progress_callback=None):
|
|
| 166 |
observer = os.getenv("VECTARA_AGENTIC_OBSERVER_TYPE", "NO_OBSERVER")
|
| 167 |
)
|
| 168 |
fallback_agent_config = AgentConfig(
|
| 169 |
-
agent_type = os.getenv("VECTARA_AGENTIC_FALLBACK_AGENT_TYPE", AgentType.
|
| 170 |
main_llm_provider = os.getenv("VECTARA_AGENTIC_FALLBACK_MAIN_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
| 171 |
main_llm_model_name = os.getenv("VECTARA_AGENTIC_FALLBACK_MAIN_MODEL_NAME", ""),
|
| 172 |
tool_llm_provider = os.getenv("VECTARA_AGENTIC_FALLBACK_TOOL_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
|
|
|
| 158 |
"""
|
| 159 |
|
| 160 |
agent_config = AgentConfig(
|
| 161 |
+
agent_type = os.getenv("VECTARA_AGENTIC_AGENT_TYPE", AgentType.FUNCTION_CALLING.value),
|
| 162 |
main_llm_provider = os.getenv("VECTARA_AGENTIC_MAIN_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
| 163 |
main_llm_model_name = os.getenv("VECTARA_AGENTIC_MAIN_MODEL_NAME", ""),
|
| 164 |
tool_llm_provider = os.getenv("VECTARA_AGENTIC_TOOL_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
|
|
|
| 166 |
observer = os.getenv("VECTARA_AGENTIC_OBSERVER_TYPE", "NO_OBSERVER")
|
| 167 |
)
|
| 168 |
fallback_agent_config = AgentConfig(
|
| 169 |
+
agent_type = os.getenv("VECTARA_AGENTIC_FALLBACK_AGENT_TYPE", AgentType.FUNCTION_CALLING.value),
|
| 170 |
main_llm_provider = os.getenv("VECTARA_AGENTIC_FALLBACK_MAIN_LLM_PROVIDER", ModelProvider.OPENAI.value),
|
| 171 |
main_llm_model_name = os.getenv("VECTARA_AGENTIC_FALLBACK_MAIN_MODEL_NAME", ""),
|
| 172 |
tool_llm_provider = os.getenv("VECTARA_AGENTIC_FALLBACK_TOOL_LLM_PROVIDER", ModelProvider.OPENAI.value),
|