Update src/embeddings_model.py

src/embeddings_model.py

@@ -12,7 +12,7 @@ class GEmbeddings(BaseEmbedding):
         self._model_name = model_name
 
     def gai_embed_content(self, text: str) -> List[float]:
-        return genai.embed_content(model=self._model_name, content=text)
+        return genai.embed_content(model=self._model_name, content=text, output_dimensionality=768)
 
     def _get_query_embedding(self, query: str) -> List[float]:
         embeddings = self.gai_embed_content(query)