mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-05 04:38:26 +00:00
[Partner] Google GenAi new release (#14882)
to support the system message merging Also fix integration tests that weren't passing
This commit is contained in:
parent
2d0f1cae8c
commit
f88af1f1cd
@ -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"
|
||||
|
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user