mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
921894960b
- Add langchain.llms.GooglePalm for text completion, - Add langchain.chat_models.ChatGooglePalm for chat completion, - Add langchain.embeddings.GooglePalmEmbeddings for sentence embeddings, - Add example field to HumanMessage and AIMessage so that users can feed in examples into the PaLM Chat API, - Add system and unit tests. Note async completion for the Text API is not yet supported and will be included in a future PR. Happy for feedback on any aspect of this PR, especially our choice of adding an example field to Human and AI Message objects to enable passing example messages to the API.
26 lines
751 B
Python
26 lines
751 B
Python
"""Test Google PaLM Text API wrapper.
|
|
|
|
Note: This test must be run with the GOOGLE_API_KEY environment variable set to a
|
|
valid API key.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
|
|
from langchain.llms.google_palm import GooglePalm
|
|
from langchain.llms.loading import load_llm
|
|
|
|
|
|
def test_google_palm_call() -> None:
|
|
"""Test valid call to Google PaLM text API."""
|
|
llm = GooglePalm(max_output_tokens=10)
|
|
output = llm("Say foo:")
|
|
assert isinstance(output, str)
|
|
|
|
|
|
def test_saving_loading_llm(tmp_path: Path) -> None:
|
|
"""Test saving/loading a Google PaLM LLM."""
|
|
llm = GooglePalm(max_output_tokens=10)
|
|
llm.save(file_path=tmp_path / "google_palm.yaml")
|
|
loaded_llm = load_llm(tmp_path / "google_palm.yaml")
|
|
assert loaded_llm == llm
|