From 52bc2fb27aa97d63f5be181849a0c7283dbbba46 Mon Sep 17 00:00:00 2001 From: Surya2k1 Date: Sun, 1 Dec 2024 20:59:45 +0530 Subject: [PATCH] Refactor embedding model name in call --- google/generativeai/types/model_types.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/google/generativeai/types/model_types.py b/google/generativeai/types/model_types.py index ff66d6339..631e44d33 100644 --- a/google/generativeai/types/model_types.py +++ b/google/generativeai/types/model_types.py @@ -355,7 +355,10 @@ def make_model_name(name: AnyModelNameOptions): if isinstance(name, (Model, protos.Model, TunedModel, protos.TunedModel)): name = name.name # pytype: disable=attribute-error elif isinstance(name, str): - name = name + if "/" not in name: + name = "models/" + name + else: + name = name else: raise TypeError( "Invalid input type. Expected one of the following types: `str`, `Model`, or `TunedModel`."