Morgan Funtowicz
commited on
Commit
·
11fa46b
1
Parent(s):
a8540ed
misc: missing deps
Browse files- requirements.txt +1 -0
- tests/test_openai.py +3 -3
requirements.txt
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
# auto-round>=0.5.0
|
2 |
loguru>=0.7.3
|
|
|
3 |
sentence-transformers
|
4 |
# transformers>=4.51
|
5 |
torch>=2.5.0
|
|
|
1 |
# auto-round>=0.5.0
|
2 |
loguru>=0.7.3
|
3 |
+
intel-extension-for-pytorch>=2.7.0
|
4 |
sentence-transformers
|
5 |
# transformers>=4.51
|
6 |
torch>=2.5.0
|
tests/test_openai.py
CHANGED
@@ -11,7 +11,7 @@ from openai.types import CreateEmbeddingResponse
|
|
11 |
def client():
|
12 |
return OpenAI(
|
13 |
base_url=os.environ.get("HFENDPOINTS_BASE_URL", "http://localhost:8000/api/v1/"),
|
14 |
-
api_key=os.environ.get("HFENDPOINTS_API_KEY", "hf_notdefined"),
|
15 |
organization="hfendpoints-images",
|
16 |
)
|
17 |
|
@@ -24,7 +24,7 @@ def test_openai_embedding_single(client, encoding_format: Literal["float", "base
|
|
24 |
encoding_format=encoding_format,
|
25 |
)
|
26 |
|
27 |
-
assert response.model
|
28 |
|
29 |
assert response.object == "list"
|
30 |
assert len(response.data) == 1
|
@@ -41,7 +41,7 @@ def test_openai_embedding_batch(client, encoding_format: Literal["float", "base6
|
|
41 |
encoding_format=encoding_format,
|
42 |
)
|
43 |
|
44 |
-
assert response.model
|
45 |
|
46 |
assert response.object == "list"
|
47 |
assert len(response.data) == 2
|
|
|
11 |
def client():
|
12 |
return OpenAI(
|
13 |
base_url=os.environ.get("HFENDPOINTS_BASE_URL", "http://localhost:8000/api/v1/"),
|
14 |
+
api_key=os.environ.get("HFENDPOINTS_API_KEY", os.environ.get("HF_TOKEN", "hf_notdefined")),
|
15 |
organization="hfendpoints-images",
|
16 |
)
|
17 |
|
|
|
24 |
encoding_format=encoding_format,
|
25 |
)
|
26 |
|
27 |
+
assert len(response.model)
|
28 |
|
29 |
assert response.object == "list"
|
30 |
assert len(response.data) == 1
|
|
|
41 |
encoding_format=encoding_format,
|
42 |
)
|
43 |
|
44 |
+
assert len(response.model)
|
45 |
|
46 |
assert response.object == "list"
|
47 |
assert len(response.data) == 2
|