Spaces:
Paused
Paused
removed debug to work with old agent setup
Browse files
main.py
CHANGED
|
@@ -42,7 +42,6 @@ def initialize_agent(
|
|
| 42 |
rag_config: Optional[RAGConfig] = None,
|
| 43 |
model_kwargs: Dict[str, Any] = {},
|
| 44 |
system_prompt: str = "MEDICAL_ASSISTANT",
|
| 45 |
-
debug: bool = False,
|
| 46 |
):
|
| 47 |
"""Initialize the MedRAX agent with specified tools and configuration.
|
| 48 |
|
|
@@ -124,7 +123,6 @@ def initialize_agent(
|
|
| 124 |
log_dir="logs",
|
| 125 |
system_prompt=prompt,
|
| 126 |
checkpointer=checkpointer,
|
| 127 |
-
debug=debug,
|
| 128 |
)
|
| 129 |
print("Agent initialized")
|
| 130 |
|
|
@@ -187,7 +185,6 @@ if __name__ == "__main__":
|
|
| 187 |
top_p=0.95,
|
| 188 |
model_kwargs=model_kwargs,
|
| 189 |
rag_config=rag_config,
|
| 190 |
-
debug=True,
|
| 191 |
system_prompt="MEDICAL_ASSISTANT",
|
| 192 |
)
|
| 193 |
|
|
|
|
| 42 |
rag_config: Optional[RAGConfig] = None,
|
| 43 |
model_kwargs: Dict[str, Any] = {},
|
| 44 |
system_prompt: str = "MEDICAL_ASSISTANT",
|
|
|
|
| 45 |
):
|
| 46 |
"""Initialize the MedRAX agent with specified tools and configuration.
|
| 47 |
|
|
|
|
| 123 |
log_dir="logs",
|
| 124 |
system_prompt=prompt,
|
| 125 |
checkpointer=checkpointer,
|
|
|
|
| 126 |
)
|
| 127 |
print("Agent initialized")
|
| 128 |
|
|
|
|
| 185 |
top_p=0.95,
|
| 186 |
model_kwargs=model_kwargs,
|
| 187 |
rag_config=rag_config,
|
|
|
|
| 188 |
system_prompt="MEDICAL_ASSISTANT",
|
| 189 |
)
|
| 190 |
|