diff options
| author | sigoden <sigoden@gmail.com> | 2024-09-24 07:42:24 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-24 07:42:24 +0800 |
| commit | 912773c25a113f49c5df63cd3a8086d38c75103e (patch) | |
| tree | 7d653edf095f0949d1eeb2f83ec3e5616972c3b1 /src/client/openai.rs | |
| parent | 00c4a6e421f01590ffc3fa5601e93d5ec755fca7 (diff) | |
| download | aichat-912773c25a113f49c5df63cd3a8086d38c75103e.tar.gz | |
refactor: embeddings/rerank fn accept ref data (#878)
Diffstat (limited to 'src/client/openai.rs')
| -rw-r--r-- | src/client/openai.rs | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs index 4876ed3..c4c2b0c 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -61,7 +61,7 @@ fn prepare_chat_completions( Ok(request_data) } -fn prepare_embeddings(self_: &OpenAIClient, data: EmbeddingsData) -> Result<RequestData> { +fn prepare_embeddings(self_: &OpenAIClient, data: &EmbeddingsData) -> Result<RequestData> { let api_key = self_.get_api_key()?; let api_base = self_ .get_api_base() @@ -294,7 +294,7 @@ pub fn openai_build_chat_completions_body(data: ChatCompletionsData, model: &Mod body } -pub fn openai_build_embeddings_body(data: EmbeddingsData, model: &Model) -> Value { +pub fn openai_build_embeddings_body(data: &EmbeddingsData, model: &Model) -> Value { json!({ "input": data.texts, "model": model.name() @@ -315,9 +315,7 @@ pub fn openai_extract_chat_completions(data: &Value) -> Result<ChatCompletionsOu call["id"].as_str(), ) { let arguments: Value = arguments.parse().with_context(|| { - format!( - "Tool call '{name}' is invalid: arguments must be in valid JSON format" - ) + format!("Tool call '{name}' is invalid: arguments must be in valid JSON format") })?; tool_calls.push(ToolCall::new( name.to_string(), |