j dZddlmZddlmZmZddlmZddlm Z m Z e ee e e fzZ GddeZGdd e Zy ) aProvider interface contracts for model layer clients. Defines the base ``ProviderABC`` abstract base class that all model provider clients (LiteLLM, Ollama, llama.cpp) must implement. This ensures structural and behavioral compatibility across providers. ) annotations)ABCabstractmethod)Iterator)AnyProtocolceZdZdZe ddZe d dZed dZed dZed dZ y) ProviderABCzIAbstract base class defining the contract for all model provider clients.c y)ajSend a chat completion request. Args: model: The model identifier to use. messages: The conversation messages. **kwargs: Additional provider-specific parameters. Returns: The model's response content as a string, or an iterator of response chunks when streaming is requested. Nselfmodelmessageskwargss Q/home/oem/CyberBrain26-unified/src/cyberbrain26/model_layer/provider_interface.pychatzProviderABC.chatc y)aCreate embeddings for the given texts. Args: model: The model identifier to use. texts: The texts to embed. **kwargs: Additional provider-specific parameters. Returns: A list of embedding vectors. Nr rrtextsrs rembedzProviderABC.embed(rrcy)zrList available models from the provider. Returns: A list of model info dictionaries. Nr rs r list_modelszProviderABC.list_models:rrcy)zCheck if the provider is available/healthy. Returns: True if the provider is reachable and operational. Nr rs r is_availablezProviderABC.is_availableBrrcy)z{Perform a health check on the provider. Returns: A dictionary with health status information. Nr rs r health_checkzProviderABC.health_checkJrrNrstrrzlist[dict[str, str]]rrreturn ChatResultrr"rz list[str]rrr#zlist[list[float]]r#zlist[dict[str, Any]]r#boolr#zdict[str, Any]) __name__ __module__ __qualname____doc__rrrrrr r rrr r sS  '      $        "      rr cXeZdZdZ ddZ d dZd dZd dZd dZy) ProviderProtocolz:Structural protocol for provider duck-typing verification.c yNr r s rrzProviderProtocol.chatV rc yr1r rs rrzProviderProtocol.embed^r2rcyr1r rs rrzProviderProtocol.list_modelsf rcyr1r rs rrzProviderProtocol.is_availableir5rcyr1r rs rr zProviderProtocol.health_checklr5rNr!r%r&r'r)) r*r+r,r-rrrrr r rrr/r/SsjD  '               rr/N)r- __future__rabcrrcollections.abcrtypingrrr"dictr$r r/r rrr=sK##$ 8DcN+ + O> #> B x r