5194fabe62
- Added `delete_classifier` function to delete a classifier by ID. - Added corresponding unit tests for the new function. - Updated the client example to demonstrate classifier deletion. - Renamed `jina_client_test.v` to `api_test.v` for better organization. - Renamed `model_embed.v` to `embeddings_api.v` for better organization. - Refactored the embedding API to use enums for task and truncate types, and added error handling for invalid inputs.
103 lines
2.7 KiB
V
103 lines
2.7 KiB
V
module jina
|
|
|
|
import time
|
|
|
|
fn setup_client() !&Jina {
|
|
mut client := get()!
|
|
return client
|
|
}
|
|
|
|
fn test_create_embeddings() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
embeddings := client.create_embeddings(
|
|
input: ['Hello', 'World']
|
|
model: .jina_embeddings_v3
|
|
task: 'separation'
|
|
) or { panic('Error while creating embeddings: ${err}') }
|
|
|
|
assert embeddings.data.len > 0
|
|
assert embeddings.object == 'list' // Check the object type
|
|
assert embeddings.model == 'jina-embeddings-v3'
|
|
}
|
|
|
|
fn test_rerank() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
rerank_result := client.rerank(
|
|
model: .reranker_v2_base_multilingual
|
|
query: 'skincare products'
|
|
documents: ['Product A', 'Product B', 'Product C']
|
|
top_n: 2
|
|
) or { panic('Error while reranking: ${err}') }
|
|
|
|
assert rerank_result.results.len == 2
|
|
assert rerank_result.model == 'jina-reranker-v2-base-multilingual'
|
|
}
|
|
|
|
fn test_train() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
train_result := client.train(
|
|
model: .jina_clip_v1
|
|
input: [
|
|
TrainingExample{
|
|
text: 'A photo of a cat'
|
|
label: 'cat'
|
|
},
|
|
TrainingExample{
|
|
text: 'A photo of a dog'
|
|
label: 'dog'
|
|
},
|
|
]
|
|
) or { panic('Error while training: ${err}') }
|
|
|
|
assert train_result.classifier_id.len > 0
|
|
assert train_result.num_samples == 2
|
|
}
|
|
|
|
fn test_classify() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
classify_result := client.classify(
|
|
model: .jina_clip_v1
|
|
input: [
|
|
ClassificationInput{
|
|
text: 'A photo of a cat'
|
|
},
|
|
ClassificationInput{
|
|
image: 'https://letsenhance.io/static/73136da51c245e80edc6ccfe44888a99/1015f/MainBefore.jpg'
|
|
},
|
|
]
|
|
labels: ['cat', 'dog']
|
|
) or { panic('Error while classifying: ${err}') }
|
|
|
|
assert classify_result.data.len == 2
|
|
assert classify_result.data[0].prediction in ['cat', 'dog']
|
|
assert classify_result.data[1].prediction in ['cat', 'dog']
|
|
assert classify_result.data[0].object == 'classification'
|
|
assert classify_result.data[1].object == 'classification'
|
|
}
|
|
|
|
fn test_get_classifiers() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
classifiers := client.list_classifiers() or { panic('Error fetching classifiers: ${err}') }
|
|
assert classifiers.len != 0
|
|
}
|
|
|
|
// Delete classifier
|
|
fn test_delete_classifiers() {
|
|
time.sleep(1 * time.second)
|
|
mut client := setup_client()!
|
|
|
|
classifiers := client.list_classifiers() or { panic('Error fetching classifiers: ${err}') }
|
|
assert classifiers.len != 0
|
|
|
|
delete_result := client.delete_classifier(classifier_id: classifiers[0].classifier_id) or {
|
|
panic('Error deleting classifier: ${err}')
|
|
}
|
|
|
|
assert delete_result == '{"message":"Classifier ${classifiers[0].classifier_id} deleted"}'
|
|
}
|