Skip to content

Commit

Permalink
Fix incorrect completion() signature for Databricks provider
Browse files Browse the repository at this point in the history
Signed-off-by: Yuan Tang <[email protected]>
  • Loading branch information
terrytangyuan committed Oct 11, 2024
1 parent 1ff0476 commit 59b4a4b
Showing 1 changed file with 8 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,14 @@ async def initialize(self) -> None:
async def shutdown(self) -> None:
pass

def completion(self, request: CompletionRequest) -> AsyncGenerator:
def completion(
self,
model: str,
content: InterleavedTextMedia,
sampling_params: Optional[SamplingParams] = SamplingParams(),
stream: Optional[bool] = False,
logprobs: Optional[LogProbConfig] = None,
) -> AsyncGenerator:
raise NotImplementedError()

def chat_completion(
Expand Down

0 comments on commit 59b4a4b

Please sign in to comment.