Fix incorrect completion() signature for Databricks provider (#236)

This commit is contained in:
Yuan Tang 2024-10-11 11:47:57 -04:00 committed by GitHub
parent 9fbe8852aa
commit 2128e61da2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 19 additions and 6 deletions

View file

@ -4,7 +4,6 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Optional
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel, Field
@ -19,4 +18,4 @@ class DatabricksImplConfig(BaseModel):
api_token: str = Field(
default=None,
description="The Databricks API token",
)
)