feat: initial implementation of snowflake provider + distro

This commit is contained in:
alejandro 2024-10-31 13:00:22 -04:00
parent 4aa1bf6a60
commit 59d856363e
5 changed files with 293 additions and 0 deletions

View file

@ -0,0 +1,21 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel, Field
@json_schema_type
class SnowflakeImplConfig(BaseModel):
url: str = Field(
default=None,
description="The URL for the Snowflake Cortex model serving endpoint",
)
api_token: str = Field(
default=None,
description="The Snowflake Cortex API token",
)