mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
9 lines
273 B
Python
9 lines
273 B
Python
|
from langchain_google_genai.llms import GoogleModelFamily
|
||
|
|
|
||
|
|
|
||
|
|
def test_model_family() -> None:
|
||
|
|
model = GoogleModelFamily("gemini-pro")
|
||
|
|
assert model == GoogleModelFamily.GEMINI
|
||
|
|
model = GoogleModelFamily("gemini-ultra")
|
||
|
|
assert model == GoogleModelFamily.GEMINI
|