(
self,
*,
payload: CreateCompletionRequest,
prompt_text: str,
prompt_plan: PromptPlan,
grammar_text: Optional[str] = None,
chat_tool_name: Optional[str] = None,
on_stream_chunk: Optional[Callable[[CompletionChunk], None]] = None,
on_done: Optional[Callable[[OpenAICompletion], None]] = None,
on_error: Optional[Callable[[BaseException], None]] = None,
)
| 15646 | ) |
| 15647 | |
| 15648 | def request_from_prepared( |
| 15649 | self, |
| 15650 | *, |
| 15651 | payload: CreateCompletionRequest, |
| 15652 | prompt_text: str, |
| 15653 | prompt_plan: PromptPlan, |
| 15654 | grammar_text: Optional[str] = None, |
| 15655 | chat_tool_name: Optional[str] = None, |
| 15656 | on_stream_chunk: Optional[Callable[[CompletionChunk], None]] = None, |
| 15657 | on_done: Optional[Callable[[OpenAICompletion], None]] = None, |
| 15658 | on_error: Optional[Callable[[BaseException], None]] = None, |
| 15659 | ) -> CompletionRequest: |
| 15660 | model = self.scheduler.model |
| 15661 | prompt_visible_start = self.prompt_visible_start(prompt_plan) |
| 15662 | return CompletionRequest.from_prepared( |
| 15663 | payload=payload, |
| 15664 | prompt_text=prompt_text, |
| 15665 | prompt_plan=prompt_plan, |
| 15666 | max_seq_len=model.max_seq_len, |
| 15667 | max_output_tokens=model.max_output_tokens, |
| 15668 | prompt_visible_start=prompt_visible_start, |
| 15669 | prompt_records=self.initial_prompt_records( |
| 15670 | payload, |
| 15671 | prompt_plan, |
| 15672 | prompt_visible_start, |
| 15673 | ), |
| 15674 | grammar_text=grammar_text, |
| 15675 | chat_tool_name=chat_tool_name, |
| 15676 | on_stream_chunk=on_stream_chunk, |
| 15677 | on_done=on_done, |
| 15678 | on_error=on_error, |
| 15679 | ) |
| 15680 | |
| 15681 | def prepare_completion_prompt( |
| 15682 | self, |
no test coverage detected