github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-05-07 03:59:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

UPdate

Browse Source
This commit is contained in:
Eugene Yurtsev
2023-05-15 14:50:03 -04:00
parent 99cfe71cd0
commit 9bd3e2e825
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -80,7 +80,7 @@ jq = {version = "^1.4.1", optional = true}
steamship = {version = "^2.16.9", optional = true}
pdfminer-six = {version = "^20221105", optional = true}
docarray = {version="^0.31.0", optional=true}
protobuf = {version="3.19", optional=true}
protobuf = {version="^3.19", optional=true}
hnswlib = {version="^0.7.0", optional=true}
lxml = {version = "^4.9.2", optional = true}
pymupdf = {version = "^1.22.3", optional = true}

16
tests/unit_tests/chat_models/test_google_palm.py
View File

@@ -14,8 +14,8 @@ from langchain.schema import (
)
@pytest.mark.requires("google.generativeai")
def test_messages_to_prompt_dict_with_valid_messages() -> None:
pytest.importorskip("google.generativeai")
result = _messages_to_prompt_dict(
[
SystemMessage(content="Prompt"),
@@ -44,8 +44,8 @@ def test_messages_to_prompt_dict_with_valid_messages() -> None:
assert result == expected
@pytest.mark.requires("google.generativeai")
def test_messages_to_prompt_dict_raises_with_misplaced_system_message() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ChatGooglePalmError) as e:
_messages_to_prompt_dict(
[
@@ -56,8 +56,8 @@ def test_messages_to_prompt_dict_raises_with_misplaced_system_message() -> None:
assert "System message must be first" in str(e)
@pytest.mark.requires("google.generativeai")
def test_messages_to_prompt_dict_raises_with_misordered_examples() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ChatGooglePalmError) as e:
_messages_to_prompt_dict(
[
@@ -68,8 +68,8 @@ def test_messages_to_prompt_dict_raises_with_misordered_examples() -> None:
assert "AI example message must be immediately preceded" in str(e)
@pytest.mark.requires("google.generativeai")
def test_messages_to_prompt_dict_raises_with_mismatched_examples() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ChatGooglePalmError) as e:
_messages_to_prompt_dict(
[
@@ -80,8 +80,8 @@ def test_messages_to_prompt_dict_raises_with_mismatched_examples() -> None:
assert "Human example message must be immediately followed" in str(e)
@pytest.mark.requires("google.generativeai")
def test_messages_to_prompt_dict_raises_with_example_after_real() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ChatGooglePalmError) as e:
_messages_to_prompt_dict(
[
@@ -93,22 +93,22 @@ def test_messages_to_prompt_dict_raises_with_example_after_real() -> None:
assert "Message examples must come before other" in str(e)
@pytest.mark.requires("google.generativeai")
def test_chat_google_raises_with_invalid_temperature() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ValueError) as e:
ChatGooglePalm(google_api_key="fake", temperature=2.0)
assert "must be in the range" in str(e)
@pytest.mark.requires("google.generativeai")
def test_chat_google_raises_with_invalid_top_p() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ValueError) as e:
ChatGooglePalm(google_api_key="fake", top_p=2.0)
assert "must be in the range" in str(e)
@pytest.mark.requires("google.generativeai")
def test_chat_google_raises_with_invalid_top_k() -> None:
pytest.importorskip("google.generativeai")
with pytest.raises(ValueError) as e:
ChatGooglePalm(google_api_key="fake", top_k=-5)
assert "must be positive" in str(e)