Commit 55ebbc3b authored by Your Name's avatar Your Name

Add debug output to model_parser showing model_name and selected parser

parent 4ee69261
...@@ -341,48 +341,70 @@ class ModelParserDispatcher: ...@@ -341,48 +341,70 @@ class ModelParserDispatcher:
def _get_parser(self) -> BaseParser: def _get_parser(self) -> BaseParser:
"""Get the appropriate parser based on model name.""" """Get the appropriate parser based on model name."""
if not self.model_name: if not self.model_name:
return ApexBig50Parser(self.tools) parser = ApexBig50Parser(self.tools)
print(f"DEBUG model_parser: model_name=None, selected parser: {type(parser).__name__}")
return parser
model_lower = self.model_name.lower() model_lower = self.model_name.lower()
# Qwen models # Qwen models
if 'qwen' in model_lower: if 'qwen' in model_lower:
return QwenParser(self.tools) parser = QwenParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: QwenParser")
return parser
# DeepSeek models # DeepSeek models
if 'deepseek' in model_lower: if 'deepseek' in model_lower:
return DeepSeekParser(self.tools) parser = DeepSeekParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: DeepSeekParser")
return parser
# Llama models # Llama models
if 'llama' in model_lower: if 'llama' in model_lower:
return LlamaParser(self.tools) parser = LlamaParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: LlamaParser")
return parser
# Mistral models # Mistral models
if 'mistral' in model_lower or 'mixtral' in model_lower: if 'mistral' in model_lower or 'mixtral' in model_lower:
return MistralParser(self.tools) parser = MistralParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: MistralParser")
return parser
# Claude models # Claude models
if 'claude' in model_lower: if 'claude' in model_lower:
return ClaudeParser(self.tools) parser = ClaudeParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: ClaudeParser")
return parser
# Command R models # Command R models
if 'command' in model_lower: if 'command' in model_lower:
return CommandRParser(self.tools) parser = CommandRParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: CommandRParser")
return parser
# Gemma models # Gemma models
if 'gemma' in model_lower: if 'gemma' in model_lower:
return GemmaParser(self.tools) parser = GemmaParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: GemmaParser")
return parser
# Grok models # Grok models
if 'grok' in model_lower: if 'grok' in model_lower:
return GrokParser(self.tools) parser = GrokParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: GrokParser")
return parser
# Phi models # Phi models
if 'phi' in model_lower: if 'phi' in model_lower:
return PhiParser(self.tools) parser = PhiParser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: PhiParser")
return parser
# Default: use catch-all parser # Default: use catch-all parser
return ApexBig50Parser(self.tools) parser = ApexBig50Parser(self.tools)
print(f"DEBUG model_parser: model_name={self.model_name}, selected parser: ApexBig50Parser (default)")
return parser
def parse(self, text: str) -> List[Dict]: def parse(self, text: str) -> List[Dict]:
"""Parse tool calls from model output.""" """Parse tool calls from model output."""
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment