phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-30 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added new models and updated chat_completin logic to follow the format

Browse source
This commit is contained in:
Sajikumar JS 2025-03-13 18:43:09 +05:30
parent 44c51efc55
commit e7b7b102cf
4 changed files with 233 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/inference/watsonx/config.py
View file

@ -8,7 +8,7 @@ import os
from typing import Optional, Dict, Any
from llama_stack.schema_utils import json_schema_type
from pydantic import BaseModel, Field
from pydantic import BaseModel, Field, SecretStr
class WatsonXProviderDataValidator(BaseModel):
@ -24,7 +24,7 @@ class WatsonXConfig(BaseModel):
default_factory=lambda: os.getenv("WATSONX_BASE_URL", "https://us-south.ml.cloud.ibm.com"),
description="A base url for accessing the Watsonx.ai",
)
api_key: Optional[str] = Field(
api_key: Optional[SecretStr] = Field(
default_factory=lambda: os.getenv("WATSONX_API_KEY"),
description="The Watsonx API key, only needed of using the hosted service",
)