Skip to content

Commit

Permalink
chore: update inference version
Browse files Browse the repository at this point in the history
  • Loading branch information
LinkW77 authored and taskingaijc committed Sep 10, 2024
1 parent 8240f3e commit db96caa
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 6 deletions.
2 changes: 1 addition & 1 deletion inference/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ class Config:

def __init__(self):
# version
self.VERSION = "v0.2.19"
self.VERSION = "v0.2.20"

# mode
self.MODE = load_str_env("MODE", required=True)
Expand Down
5 changes: 1 addition & 4 deletions inference/test/test_rerank.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,10 +79,7 @@ async def test_rerank(self, test_data):
for result in results:
assert result.get("document").get("text") in self.documents
assert result.get("relevance_score") >= 0.0
if "jina-colbert-v1-en" in model_schema_id:
assert result.get("relevance_score") <= 10.0
else:
assert result.get("relevance_score") <= 1.0
assert result.get("relevance_score") <= 1.0
assert result.get("index") == self.documents.index(result.get("document").get("text"))

@pytest.mark.test_id("inference_032")
Expand Down
2 changes: 2 additions & 0 deletions inference/test/utils/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,8 @@
"mistralai/mistral-small-latest",
"groq/llama-3.1-8b-instant",
"groq/gemma-7b",
"reka/reka-edge", # This model temporarily encountered an internal server error in inference version 0.2.19
"jina/jina-colbert-v1-en", # The relevance_score returned by this model lacks a defined range
]

load_dotenv()
Expand Down
2 changes: 1 addition & 1 deletion inference/test/utils/wildcard_test_cases.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ wildcard_test_cases:
- provider_id: "openrouter"
cases:
- model_schema_id: "openrouter/wildcard"
provider_model_id: "nousresearch/nous-capybara-7b:free"
provider_model_id: "mattshumer/reflection-70b:free"
model_type: "chat_completion"
streaming: True
function_call: True
Expand Down

0 comments on commit db96caa

Please sign in to comment.