j.ddlZddlmZmZddlmZddlmZddlm Z ddl m Z m Z ejejd ZGd d ZGd d Zy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)OllamaConnectionTestResponse) OllamaModel)AsyncRawOllamaClientRawOllamaClient.c eZdZdefdZedefdZeddde de je d e je de jefd Zeddde de je d e je defd Zy) OllamaClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs t/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/ollama/client.py__init__zOllamaClient.__init__s*.Ireturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawOllamaClient rrs rwith_raw_responsezOllamaClient.with_raw_responserNapi_keyrequest_optionsbase_urlrrcV|jj|||}|jS)a Fetches the list of models available from the Ollama instance. URL may be provided with or without /v1 suffix (e.g., http://localhost:11434 or http://localhost:11434/v1). The /v1 suffix will be automatically removed for model discovery. For actual LLM inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. Parameters ---------- base_url : str Base URL of the Ollama instance. May include /v1 suffix which will be automatically removed for connection testing. For inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. api_key : typing.Optional[str] Optional API key for authenticated Ollama instances. If provided, will be sent as Bearer token in Authorization header. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.List[OllamaModel] Models retrieved successfully Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.ollama.list_models(base_url='http://localhost:11434/v1', ) r rrr list_modelsdatarr rr _responses rr$zOllamaClient.list_modelss-@$$00(Get0u ~~rcV|jj|||}|jS)a Validates that the provided Ollama URL is reachable. URL may be provided with or without /v1 suffix (e.g., http://localhost:11434 or http://localhost:11434/v1). The /v1 suffix will be automatically removed for connection testing. For inference, use the URL with /v1 suffix. Parameters ---------- base_url : str Base URL of the Ollama instance. May include /v1 suffix which will be automatically removed for connection testing. For inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. api_key : typing.Optional[str] Optional API key for authenticated Ollama instances. If provided, will be sent as Bearer token in Authorization header. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OllamaConnectionTestResponse Connection test successful Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.ollama.test_connection(base_url='http://localhost:11434/v1', ) r"rtest_connectionr%r&s rr*zOllamaClient.test_connectionAs3@$$44w5 ~~r)__name__ __module__ __qualname__rrpropertyr rOMITstrtypingOptionalrListrr$rr*rrr r sJ*;J ?  )-;? !!% !  8 ! [ ! !N)-;? ##% #  8 # & #rr c eZdZdefdZedefdZeddde de je d e je de jefd Zeddde de je d e je defd Zy) AsyncOllamaClientrc&t||_yr)r rrs rrzAsyncOllamaClient.__init__hs/~Nrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawOllamaClient rrs rrz#AsyncOllamaClient.with_raw_responsekrrNrr rrcrK|jj|||d{}|jS7w)a. Fetches the list of models available from the Ollama instance. URL may be provided with or without /v1 suffix (e.g., http://localhost:11434 or http://localhost:11434/v1). The /v1 suffix will be automatically removed for model discovery. For actual LLM inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. Parameters ---------- base_url : str Base URL of the Ollama instance. May include /v1 suffix which will be automatically removed for connection testing. For inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. api_key : typing.Optional[str] Optional API key for authenticated Ollama instances. If provided, will be sent as Bearer token in Authorization header. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- typing.List[OllamaModel] Models retrieved successfully Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.ollama.list_models(base_url='http://localhost:11434/v1', ) asyncio.run(main()) r"Nr#r&s rr$zAsyncOllamaClient.list_modelsvsAF**66w7  ~~ "757crK|jj|||d{}|jS7w)a Validates that the provided Ollama URL is reachable. URL may be provided with or without /v1 suffix (e.g., http://localhost:11434 or http://localhost:11434/v1). The /v1 suffix will be automatically removed for connection testing. For inference, use the URL with /v1 suffix. Parameters ---------- base_url : str Base URL of the Ollama instance. May include /v1 suffix which will be automatically removed for connection testing. For inference, use the URL with /v1 suffix for OpenAI-compatible endpoints. api_key : typing.Optional[str] Optional API key for authenticated Ollama instances. If provided, will be sent as Bearer token in Authorization header. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- OllamaConnectionTestResponse Connection test successful Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.ollama.test_connection(base_url='http://localhost:11434/v1', ) asyncio.run(main()) r"Nr)r&s rr*z!AsyncOllamaClient.test_connectionsAF**::w;  ~~ r:)r+r,r-rrr.r rr/r0r1r2rr3rr$rr*r4rrr6r6gsO*<O #7  )-;? &&% &  8 & [ ! &X)-;? &&% &  8 & & &rr6)r1core.client_wrapperrrcore.request_optionsr%types.ollama_connection_test_responsertypes.ollama_modelr raw_clientr r castAnyr/r r6r4rrrCsJG1P,=v{{6::s#UUp]]r