github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 05:43:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/delete bad code (#253)

Browse Source
This commit is contained in:
Harrison Chase 2022-12-03 13:28:29 -08:00 committed by GitHub
parent db58032973
commit ac2c2f6f28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
langchain/model_laboratory.py
View File

@ -1,9 +1,8 @@
"""Experiment with different models."""
from __future__ import annotations
from typing import List, Optional, Sequence, Union
from typing import List, Optional, Sequence
from langchain.agents.agent import Agent
from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
from langchain.input import get_color_mapping, print_text
@ -14,22 +13,19 @@ from langchain.prompts.prompt import PromptTemplate
class ModelLaboratory:
"""Experiment with different models."""
def __init__(
self, chains: Sequence[Union[Chain, Agent]], names: Optional[List[str]] = None
):
def __init__(self, chains: Sequence[Chain], names: Optional[List[str]] = None):
"""Initialize with chains to experiment with.
Args:
chains: list of chains to experiment with.
"""
for chain in chains:
if not isinstance(chain, (Chain, Agent)):
if not isinstance(chain, Chain):
raise ValueError(
"ModelLaboratory should now be initialized with Chains or Agents. "
"ModelLaboratory should now be initialized with Chains. "
"If you want to initialize with LLMs, use the `from_llms` method "
"instead (`ModelLaboratory.from_llms(...)`)"
)
if isinstance(chain, Chain):
if len(chain.input_keys) != 1:
raise ValueError(
"Currently only support chains with one input variable, "