Allows different models for text-embedding
This commit is contained in:
@@ -33,7 +33,9 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|||||||
],
|
],
|
||||||
};
|
};
|
||||||
|
|
||||||
let result = gemini.text_embeddings(&embedding_request).await?;
|
let result = gemini
|
||||||
|
.text_embeddings(&embedding_request, "textembedding-gecko@003")
|
||||||
|
.await?;
|
||||||
println!("Response: {:?}", result);
|
println!("Response: {:?}", result);
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@@ -185,8 +185,8 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
|
|||||||
pub async fn text_embeddings(
|
pub async fn text_embeddings(
|
||||||
&self,
|
&self,
|
||||||
request: &TextEmbeddingRequest,
|
request: &TextEmbeddingRequest,
|
||||||
|
model: &str,
|
||||||
) -> Result<TextEmbeddingResponse> {
|
) -> Result<TextEmbeddingResponse> {
|
||||||
let model = "models/embedding-001";
|
|
||||||
let endpoint_url = format!(
|
let endpoint_url = format!(
|
||||||
"https://{}/v1/projects/{}/locations/{}/publishers/google/models/{}:predict",
|
"https://{}/v1/projects/{}/locations/{}/publishers/google/models/{}:predict",
|
||||||
self.api_endpoint, self.project_id, self.location_id, model,
|
self.api_endpoint, self.project_id, self.location_id, model,
|
||||||
|
|||||||
Reference in New Issue
Block a user