mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-07 14:26:44 +00:00
fix for model in OpenAIEmbeddingsResponse and rebase
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
15042c6c31
commit
179399edf4
2 changed files with 9 additions and 1 deletions
|
@ -419,9 +419,10 @@ class OllamaInferenceAdapter(
|
|||
prompt_tokens=response.usage.prompt_tokens,
|
||||
total_tokens=response.usage.total_tokens,
|
||||
)
|
||||
# TODO: Investigate why model_obj.identifier is used instead of response.model
|
||||
return OpenAIEmbeddingsResponse(
|
||||
data=data,
|
||||
model=response.model,
|
||||
model=model_obj.identifier,
|
||||
usage=usage,
|
||||
)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue