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

Allow setting environment variables from llama stack run and fix ollama

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-17 19:33:48 -08:00
parent a061f3f8c1
commit b1d119466e
19 changed files with 129 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/providers/remote/inference/ollama/config.py
View file

@ -6,17 +6,17 @@
from typing import Any, Dict
from llama_stack.distribution.datatypes import RemoteProviderConfig
from pydantic import BaseModel
DEFAULT_OLLAMA_PORT = 11434
DEFAULT_OLLAMA_URL = "http://localhost:11434"
class OllamaImplConfig(RemoteProviderConfig):
port: int
class OllamaImplConfig(BaseModel):
url: str = DEFAULT_OLLAMA_URL
@classmethod
def sample_run_config(
cls, port_str: str = str(DEFAULT_OLLAMA_PORT)
cls, url: str = "${env.OLLAMA_URL:http://localhost:11434}", **kwargs
) -> Dict[str, Any]:
return {"port": port_str}
return {"url": url}

2
llama_stack/providers/remote/inference/ollama/ollama.py
View file

@ -82,7 +82,7 @@ class OllamaInferenceAdapter(Inference, ModelsProtocolPrivate):
return AsyncClient(host=self.url)
async def initialize(self) -> None:
print("Initializing Ollama, checking connectivity to server...")
print(f"checking connectivity to Ollama at `{self.url}`...")
try:
await self.client.ps()
except httpx.ConnectError as e: