davidkhala.ai 0.0.9__py3-none-any.whl → 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. davidkhala/ai/api/siliconflow.py +4 -3
  2. {davidkhala_ai-0.0.9.dist-info → davidkhala_ai-0.1.0.dist-info}/METADATA +1 -1
  3. {davidkhala_ai-0.0.9.dist-info → davidkhala_ai-0.1.0.dist-info}/RECORD +4 -4
  4. {davidkhala_ai-0.0.9.dist-info → davidkhala_ai-0.1.0.dist-info}/WHEEL +0 -0
davidkhala/ai/api/siliconflow.py CHANGED
@@ -48,7 +48,7 @@ class SiliconFlow(API):
48
48
  response = self.request(f"{self.base_url}/embeddings", "POST", json=json)
49
49
  return [_['embedding'] for _ in response['data']]
50
50
 
51
- def which(self, query: str, documents: list[str], **kwargs):
51
+ def which(self, query: str, documents: list[str], **kwargs)->tuple[str,int]:
52
52
  json = {
53
53
  'model': self.model,
54
54
  'query': query,
@@ -56,6 +56,7 @@ class SiliconFlow(API):
56
56
  **kwargs
57
57
  }
58
58
  response = self.request(f"{self.base_url}/rerank", "POST", json=json)
59
- most_relevant = max(response['results'], key=lambda x: x['relevance_score'])
60
- return documents[most_relevant['index']]
59
+ most_relevant_index = max(response['results'], key=lambda x: x['relevance_score'])['index']
60
+
61
+ return documents[most_relevant_index], most_relevant_index
61
62
 
{davidkhala_ai-0.0.9.dist-info → davidkhala_ai-0.1.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: davidkhala.ai
3
- Version: 0.0.9
3
+ Version: 0.1.0
4
4
  Summary: misc AI modules
5
5
  Requires-Python: >=3.13
6
6
  Provides-Extra: ali
{davidkhala_ai-0.0.9.dist-info → davidkhala_ai-0.1.0.dist-info}/RECORD RENAMED
@@ -7,7 +7,7 @@ davidkhala/ai/ali/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
7
7
  davidkhala/ai/ali/dashscope.py,sha256=SZIzRhVHlLx3s5I2RNUh2-u8OoSdrbvoN5e1k8Mh8N0,1943
8
8
  davidkhala/ai/api/__init__.py,sha256=q2Ro5nhW5kJx2CYR1MRVamjTT5tTexPZwhrS2hwAvFM,1319
9
9
  davidkhala/ai/api/openrouter.py,sha256=khccJr5cBnudFy6Jc2O3A1TNCuHH_5W6Q2tXrkwlUYE,2308
10
- davidkhala/ai/api/siliconflow.py,sha256=uTwkJHfGZ4zF2qIRf88kQeSlm6t15izSuJOYi2SmbCs,2088
10
+ davidkhala/ai/api/siliconflow.py,sha256=JbnOSv8LJLtwYSNNB8_SMBMQzOgHDtQYZKA9A2BC4sY,2139
11
11
  davidkhala/ai/google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
12
12
  davidkhala/ai/google/adk.py,sha256=QwxYoOzT2Hol03V4NM0PF_HAzUGb4fB18VUAYacYbAY,657
13
13
  davidkhala/ai/google/gemini.py,sha256=Xf4HDOOcK4-jEBERzuLnQNFsU61P2fFx4K0z-ijvNHE,214
@@ -16,6 +16,6 @@ davidkhala/ai/huggingface/__init__.py,sha256=FJyU8eOfWQWKAvkIa5qwubF9ghsSQ8C0e6p
16
16
  davidkhala/ai/openai/__init__.py,sha256=UalrIvniAFSbCAQeCrDgBVEWCFTCPxFnoOAGVnVWkkI,1864
17
17
  davidkhala/ai/openai/azure.py,sha256=jAtZNW1d8ub3odhB_f1MXxM7MImg64MXDnqLm1-Idkk,828
18
18
  davidkhala/ai/openai/native.py,sha256=-XuHn0KSQnuFdJ4jTwZla1ts5iOW2L2Hosfo8o8iip8,257
19
- davidkhala_ai-0.0.9.dist-info/METADATA,sha256=WkyphTNgeBf8G2aJk5XgCcE7EOIlpnzIZu3d8xoG_Ss,877
20
- davidkhala_ai-0.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
21
- davidkhala_ai-0.0.9.dist-info/RECORD,,
19
+ davidkhala_ai-0.1.0.dist-info/METADATA,sha256=MnG5FylnaOFamw3sjPcMMVCUoNEtMWOfxnoyqP1A40Q,877
20
+ davidkhala_ai-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
21
+ davidkhala_ai-0.1.0.dist-info/RECORD,,