Merge pull request #10 from feloy/fix/add-missing-method

fix: openai_embeddings method added in llama-stack 0.2.10

Author: feloy

src/podman_ai_lab_stack/podman_ai_lab.py

@@ -37,6 +37,7 @@
     Message,
     OpenAIChatCompletion,
     OpenAIChatCompletionChunk,
+    OpenAIEmbeddingsResponse,
     OpenAIMessageParam,
     OpenAIResponseFormatParam,
     ResponseFormat,
@@ -64,7 +65,6 @@
     process_completion_stream_response,
 )
 from llama_stack.providers.utils.inference.prompt_adapter import (
-    chat_completion_request_to_prompt,
     completion_request_to_prompt,
     convert_image_content_to_url,
     request_has_media,
@@ -307,6 +307,16 @@ async def openai_chat_completion(
     ) -> Union[OpenAIChatCompletion, AsyncIterator[OpenAIChatCompletionChunk]]:
         pass
 
+    async def openai_embeddings(
+        self,
+        model: str,
+        input: str | list[str],
+        encoding_format: str | None = "float",
+        dimensions: int | None = None,
+        user: str | None = None,
+    ) -> OpenAIEmbeddingsResponse:
+        raise NotImplementedError()
+
     async def openai_completion(
         self,
         model: str,