github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-05 04:38:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Partner] Google GenAi new release (#14882)

Browse Source 
to support the system message merging

Also fix integration tests that weren't passing
This commit is contained in:
William FH 2023-12-18 18:35:57 -08:00 committed by GitHub
parent 2d0f1cae8c
commit f88af1f1cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/partners/google-genai/pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "langchain-google-genai" name = "langchain-google-genai"
version = "0.0.4" version = "0.0.5"
description = "An integration package connecting Google's genai package and LangChain" description = "An integration package connecting Google's genai package and LangChain"
authors = [] authors = []
readme = "README.md" readme = "README.md"

9
libs/partners/google-genai/tests/integration_tests/test_llms.py
View File

@ -9,7 +9,7 @@ from langchain_core.outputs import LLMResult
from langchain_google_genai.llms import GoogleGenerativeAI 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( @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: def test_google_generativeai_generate(model_name: str) -> None:
n = 1 if model_name == "gemini-pro" else 2 n = 1 if model_name == "gemini-pro" else 2
if model_name: llm = GoogleGenerativeAI(temperature=0.3, n=n, model=model_name)
llm = GoogleGenerativeAI(temperature=0.3, n=n, model=model_name)
else:
llm = GoogleGenerativeAI(temperature=0.3, n=n)
output = llm.generate(["Say foo:"]) output = llm.generate(["Say foo:"])
assert isinstance(output, LLMResult) assert isinstance(output, LLMResult)
assert len(output.generations) == 1 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: 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?") output = llm.get_num_tokens("How are you?")
assert output == 4 assert output == 4