community[patch]: YandexGPT Use recent yandexcloud sdk version (#19341)

Fixed inability to work with [yandexcloud
SDK](https://pypi.org/project/yandexcloud/) version higher 0.265.0

docs/docs/integrations/chat/yandex.ipynb

@@ -90,7 +90,7 @@
     }
    ],
    "source": [
-    "answer = chat_model(\n",
+    "answer = chat_model.invoke(\n",
     "    [\n",
     "        SystemMessage(\n",
     "            content=\"You are a helpful assistant that translates English to French.\"\n",

docs/docs/integrations/llms/yandex.ipynb

@@ -96,7 +96,7 @@
    "source": [
     "country = \"Russia\"\n",
     "\n",
-    "llm_chain.run(country)"
+    "llm_chain.invoke(country)"
    ]
   }
  ],

libs/community/langchain_community/chat_models/yandex.py

@@ -127,6 +127,19 @@ def _make_request(
     try:
         import grpc
         from google.protobuf.wrappers_pb2 import DoubleValue, Int64Value
+
+        try:
+            from yandex.cloud.ai.foundation_models.v1.text_common_pb2 import (
+                CompletionOptions,
+                Message,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2 import (  # noqa: E501
+                CompletionRequest,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2_grpc import (  # noqa: E501
+                TextGenerationServiceStub,
+            )
+        except ModuleNotFoundError:
             from yandex.cloud.ai.foundation_models.v1.foundation_models_pb2 import (
                 CompletionOptions,
                 Message,
@@ -166,6 +179,20 @@ async def _amake_request(self: ChatYandexGPT, messages: List[BaseMessage]) -> st
 
         import grpc
         from google.protobuf.wrappers_pb2 import DoubleValue, Int64Value
+
+        try:
+            from yandex.cloud.ai.foundation_models.v1.text_common_pb2 import (
+                CompletionOptions,
+                Message,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2 import (  # noqa: E501
+                CompletionRequest,
+                CompletionResponse,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2_grpc import (  # noqa: E501
+                TextGenerationAsyncServiceStub,
+            )
+        except ModuleNotFoundError:
             from yandex.cloud.ai.foundation_models.v1.foundation_models_pb2 import (
                 CompletionOptions,
                 Message,

libs/community/langchain_community/embeddings/__init__.py

@@ -80,6 +80,7 @@ _module_lookup = {
     "VolcanoEmbeddings": "langchain_community.embeddings.volcengine",
     "VoyageEmbeddings": "langchain_community.embeddings.voyageai",
     "XinferenceEmbeddings": "langchain_community.embeddings.xinference",
+    "YandexGPTEmbeddings": "langchain_community.embeddings.yandex",
 }
 
 

libs/community/langchain_community/embeddings/yandex.py

@@ -149,6 +149,15 @@ def _embed_with_retry(llm: YandexGPTEmbeddings, **kwargs: Any) -> Any:
 def _make_request(self: YandexGPTEmbeddings, texts: List[str]):  # type: ignore[no-untyped-def]
     try:
         import grpc
+
+        try:
+            from yandex.cloud.ai.foundation_models.v1.embedding.embedding_service_pb2 import (  # noqa: E501
+                TextEmbeddingRequest,
+            )
+            from yandex.cloud.ai.foundation_models.v1.embedding.embedding_service_pb2_grpc import (  # noqa: E501
+                EmbeddingsServiceStub,
+            )
+        except ModuleNotFoundError:
             from yandex.cloud.ai.foundation_models.v1.foundation_models_service_pb2 import (  # noqa: E501
                 TextEmbeddingRequest,
             )

libs/community/langchain_community/llms/yandex.py

@@ -186,6 +186,19 @@ def _make_request(
     try:
         import grpc
         from google.protobuf.wrappers_pb2 import DoubleValue, Int64Value
+
+        try:
+            from yandex.cloud.ai.foundation_models.v1.text_common_pb2 import (
+                CompletionOptions,
+                Message,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2 import (  # noqa: E501
+                CompletionRequest,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2_grpc import (  # noqa: E501
+                TextGenerationServiceStub,
+            )
+        except ModuleNotFoundError:
             from yandex.cloud.ai.foundation_models.v1.foundation_models_pb2 import (
                 CompletionOptions,
                 Message,
@@ -222,6 +235,20 @@ async def _amake_request(self: YandexGPT, prompt: str) -> str:
 
         import grpc
         from google.protobuf.wrappers_pb2 import DoubleValue, Int64Value
+
+        try:
+            from yandex.cloud.ai.foundation_models.v1.text_common_pb2 import (
+                CompletionOptions,
+                Message,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2 import (  # noqa: E501
+                CompletionRequest,
+                CompletionResponse,
+            )
+            from yandex.cloud.ai.foundation_models.v1.text_generation.text_generation_service_pb2_grpc import (  # noqa: E501
+                TextGenerationAsyncServiceStub,
+            )
+        except ModuleNotFoundError:
             from yandex.cloud.ai.foundation_models.v1.foundation_models_pb2 import (
                 CompletionOptions,
                 Message,

libs/community/tests/unit_tests/embeddings/test_imports.py

@@ -65,6 +65,7 @@ EXPECTED_ALL = [
     "QuantizedBiEncoderEmbeddings",
     "NeMoEmbeddings",
     "SparkLLMTextEmbeddings",
+    "YandexGPTEmbeddings",
 ]