Google model in embeddings as well as browser model enabled. (#351)
* gemini embeddings as well as gemini in browser models works now
npnpatidar committed
Mar 31, 2025 at 07:09 UTC
52dd746ce11e0ac4807ec03ba741aa2d5dd7577b
1 file changed
+3
-3
models.py
+3
-3
@@ -20,7 +20,7 @@ from langchain_huggingface import (
20
HuggingFaceEndpoint,
21
)
22
from langchain_google_genai import (
23
- GoogleGenerativeAI,
23
+ ChatGoogleGenerativeAI,
24
HarmBlockThreshold,
25
HarmCategory,
26
embeddings as google_embeddings,
@@ -267,7 +267,7 @@ def get_google_chat(
267
):
268
if not api_key:
269
api_key = get_api_key("google")
270
- return GoogleGenerativeAI(model=model_name, google_api_key=api_key, safety_settings={HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE}, **kwargs) # type: ignore
270
+ return ChatGoogleGenerativeAI(model=model_name, google_api_key=api_key, safety_settings={HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE}, **kwargs) # type: ignore
271
272
273
def get_google_embedding(
@@ -277,7 +277,7 @@ def get_google_embedding(
277
):
278
if not api_key:
279
api_key = get_api_key("google")
280
- return google_embeddings.GoogleGenerativeAIEmbeddings(model=model_name, api_key=api_key, **kwargs) # type: ignore
280
+ return google_embeddings.GoogleGenerativeAIEmbeddings(model=model_name, google_api_key=api_key, **kwargs) # type: ignore
281
282
283
# Mistral models