added consistency in embed model names

Author: bhakarboy01

google/generativeai/embedding.py

@@ -28,7 +28,7 @@
 from google.generativeai.types import text_types
 from google.generativeai.types import content_types
 
-DEFAULT_EMB_MODEL = "models/embedding-001"
+DEFAULT_EMB_MODEL = "embedding-001"
 EMBEDDING_MAX_BATCH_SIZE = 100
 
 EmbeddingTaskType = protos.TaskType

samples/embed.py

@@ -22,7 +22,7 @@ def test_embed_content(self):
 
         text = "Hello World!"
         result = genai.embed_content(
-            model="models/text-embedding-004", content=text, output_dimensionality=10
+            model="text-embedding-004", content=text, output_dimensionality=10
         )
         print(result["embedding"])
         # [END embed_content]
@@ -37,7 +37,7 @@ def batch_embed_contents(self):
             "How does the brain work?",
         ]
         result = genai.embed_content(
-            model="models/text-embedding-004", content=texts, output_dimensionality=10
+            model="text-embedding-004", content=texts, output_dimensionality=10
         )
         print(result)
         # [END batch_embed_contents]

tests/test_embedding.py

@@ -26,7 +26,7 @@
 from absl.testing import absltest
 from absl.testing import parameterized
 
-DEFAULT_EMB_MODEL = "models/embedding-001"
+DEFAULT_EMB_MODEL = "embedding-001"
 
 
 class UnitTests(parameterized.TestCase):

tests/test_embedding_async.py

@@ -26,7 +26,7 @@
 from absl.testing import absltest
 from absl.testing import parameterized
 
-DEFAULT_EMB_MODEL = "models/embedding-001"
+DEFAULT_EMB_MODEL = "embedding-001"
 
 
 class AsyncTests(parameterized.TestCase, unittest.IsolatedAsyncioTestCase):