summaryrefslogtreecommitdiffstats
path: root/src/client/ernie.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/ernie.rs')
-rw-r--r--src/client/ernie.rs8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs
index fe98ff5..d0fe7b5 100644
--- a/src/client/ernie.rs
+++ b/src/client/ernie.rs
@@ -41,7 +41,7 @@ impl Client for ErnieClient {
) -> Result<ChatCompletionsOutput> {
prepare_access_token(self, client).await?;
let request_data = prepare_chat_completions(self, data)?;
- let builder = self.request_builder(client, request_data, ApiType::ChatCompletions);
+ let builder = self.request_builder(client, request_data);
chat_completions(builder, &self.model).await
}
@@ -53,7 +53,7 @@ impl Client for ErnieClient {
) -> Result<()> {
prepare_access_token(self, client).await?;
let request_data = prepare_chat_completions(self, data)?;
- let builder = self.request_builder(client, request_data, ApiType::ChatCompletions);
+ let builder = self.request_builder(client, request_data);
chat_completions_streaming(builder, handler, &self.model).await
}
@@ -64,7 +64,7 @@ impl Client for ErnieClient {
) -> Result<EmbeddingsOutput> {
prepare_access_token(self, client).await?;
let request_data = prepare_embeddings(self, data)?;
- let builder = self.request_builder(client, request_data, ApiType::Embeddings);
+ let builder = self.request_builder(client, request_data);
embeddings(builder, &self.model).await
}
@@ -75,7 +75,7 @@ impl Client for ErnieClient {
) -> Result<RerankOutput> {
prepare_access_token(self, client).await?;
let request_data = prepare_rerank(self, data)?;
- let builder = self.request_builder(client, request_data, ApiType::Rerank);
+ let builder = self.request_builder(client, request_data);
rerank(builder, &self.model).await
}
}