phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

test(client-sdk): Update embedding test types to use latest imports (#1203)

Browse source 
# What does this PR do?
- Updates ImageContentItemImageURL import
- fixes `embedding_dimensions` metadata param

## Test Plan
- Ran pytest locally, verified embedding tests pass with new types

![Screenshot 2025-02-21 at 6 54
27 PM](https://github.com/user-attachments/assets/f80e3785-04c3-415e-9276-88aa8136bf00)

cc: @dglogo @sumitb
This commit is contained in:
Rashmi Pawar 2025-02-21 21:39:17 +05:30 • committed by GitHub
parent 46da187c07
commit da9f0b7869
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/nvidia/models.py
View file

@ -52,7 +52,7 @@ _MODEL_ENTRIES = [
provider_model_id="baai/bge-m3", provider_model_id="baai/bge-m3",
model_type=ModelType.embedding, model_type=ModelType.embedding,
metadata={ metadata={
"embedding_dimensions": 1024, "embedding_dimension": 1024,
"context_length": 8192, "context_length": 8192,
}, },
), ),

2
llama_stack/templates/nvidia/run.yaml
View file

@ -136,7 +136,7 @@ models:
provider_model_id: meta/llama-3.2-90b-vision-instruct provider_model_id: meta/llama-3.2-90b-vision-instruct
model_type: llm model_type: llm
- metadata: - metadata:
embedding_dimensions: 1024 embedding_dimension: 1024
context_length: 8192 context_length: 8192
model_id: baai/bge-m3 model_id: baai/bge-m3
provider_id: nvidia provider_id: nvidia