From f88af1f1cd62cf5a167c2a6334f4192db23669d1 Mon Sep 17 00:00:00 2001 From: William FH <13333726+hinthornw@users.noreply.github.com> Date: Mon, 18 Dec 2023 18:35:57 -0800 Subject: [PATCH] [Partner] Google GenAi new release (#14882) to support the system message merging Also fix integration tests that weren't passing --- libs/partners/google-genai/pyproject.toml | 2 +- .../google-genai/tests/integration_tests/test_llms.py | 9 +++------ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/libs/partners/google-genai/pyproject.toml b/libs/partners/google-genai/pyproject.toml index b408438e257..7769d2842dd 100644 --- a/libs/partners/google-genai/pyproject.toml +++ b/libs/partners/google-genai/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langchain-google-genai" -version = "0.0.4" +version = "0.0.5" description = "An integration package connecting Google's genai package and LangChain" authors = [] readme = "README.md" diff --git a/libs/partners/google-genai/tests/integration_tests/test_llms.py b/libs/partners/google-genai/tests/integration_tests/test_llms.py index f8985d99886..09b17d8201e 100644 --- a/libs/partners/google-genai/tests/integration_tests/test_llms.py +++ b/libs/partners/google-genai/tests/integration_tests/test_llms.py @@ -9,7 +9,7 @@ from langchain_core.outputs import LLMResult from langchain_google_genai.llms import GoogleGenerativeAI -model_names = [None, "models/text-bison-001", "gemini-pro"] +model_names = ["models/text-bison-001", "gemini-pro"] @pytest.mark.parametrize( @@ -37,10 +37,7 @@ def test_google_generativeai_call(model_name: str) -> None: ) def test_google_generativeai_generate(model_name: str) -> None: n = 1 if model_name == "gemini-pro" else 2 - if model_name: - llm = GoogleGenerativeAI(temperature=0.3, n=n, model=model_name) - else: - llm = GoogleGenerativeAI(temperature=0.3, n=n) + llm = GoogleGenerativeAI(temperature=0.3, n=n, model=model_name) output = llm.generate(["Say foo:"]) assert isinstance(output, LLMResult) assert len(output.generations) == 1 @@ -48,7 +45,7 @@ def test_google_generativeai_generate(model_name: str) -> None: def test_google_generativeai_get_num_tokens() -> None: - llm = GoogleGenerativeAI() + llm = GoogleGenerativeAI(model="models/text-bison-001") output = llm.get_num_tokens("How are you?") assert output == 4