VictorLJZ commited on
Commit
ec85157
·
1 Parent(s): 6db0a72

removed debug to work with old agent setup

Browse files
Files changed (1) hide show
  1. main.py +0 -3
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