phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki 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",
model_type=ModelType.embedding,
metadata={
"embedding_dimensions": 1024,
"embedding_dimension": 1024,
"context_length": 8192,
},
),