kovsbo commited on
Commit
f8df8fa
·
verified ·
1 Parent(s): 4035a71

Upload openpipe_llama_dual.py

Browse files
Files changed (1) hide show
  1. openpipe_llama_dual.py +4 -3
openpipe_llama_dual.py CHANGED
@@ -27,9 +27,10 @@ class OpenPipeLlamaDualParser(ToolParser):
27
  VARIANT_PIPELINE3 = "pipeline3"
28
  VARIANT_OFFICIAL = "official"
29
 
30
- def __init__(self, tokenizer: TokenizerLike):
31
- super().__init__(tokenizer)
32
  self.tokenizer = tokenizer
 
33
 
34
  def _get_template_variant(self, request: ChatCompletionRequest) -> Optional[str]:
35
  kwargs = getattr(request, "chat_template_kwargs", None)
@@ -261,4 +262,4 @@ class OpenPipeLlamaDualParser(ToolParser):
261
  tools_called=False,
262
  tool_calls=[],
263
  content=model_output,
264
- )
 
27
  VARIANT_PIPELINE3 = "pipeline3"
28
  VARIANT_OFFICIAL = "official"
29
 
30
+ def __init__(self, tokenizer: TokenizerLike, tools):
31
+ super().__init__(tokenizer, tools)
32
  self.tokenizer = tokenizer
33
+ self.tools = tools
34
 
35
  def _get_template_variant(self, request: ChatCompletionRequest) -> Optional[str]:
36
  kwargs = getattr(request, "chat_template_kwargs", None)
 
262
  tools_called=False,
263
  tool_calls=[],
264
  content=model_output,
265
+ )