Morgan Funtowicz commited on
Commit
11fa46b
·
1 Parent(s): a8540ed

misc: missing deps

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. 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 == "all-MiniLM-L6-v2"
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 == "all-MiniLM-L6-v2"
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