langchain/cookbook/optimization.ipynb

Optimization

This notebook goes over how to optimize chains using LangChain and LangSmith.

Set up

We will set an environment variable for LangSmith, and load the relevant data

In [1]:

import os

os.environ["LANGSMITH_PROJECT"] = "movie-qa"

In [2]:

import pandas as pd

In [17]:

df = pd.read_csv("data/imdb_top_1000.csv")

In [4]:

df["Released_Year"] = df["Released_Year"].astype(int, errors="ignore")

Create the initial retrieval chain

We will use a self-query retriever

In [5]:

from langchain.schema import Document
from langchain_chroma import Chroma
from langchain_openai import OpenAIEmbeddings

embeddings = OpenAIEmbeddings()

In [6]:

records = df.to_dict("records")
documents = [Document(page_content=d["Overview"], metadata=d) for d in records]

In [7]:

vectorstore = Chroma.from_documents(documents, embeddings)

In [9]:

from langchain.chains.query_constructor.base import AttributeInfo
from langchain.retrievers.self_query.base import SelfQueryRetriever
from langchain_openai import ChatOpenAI

metadata_field_info = [
    AttributeInfo(
        name="Released_Year",
        description="The year the movie was released",
        type="int",
    ),
    AttributeInfo(
        name="Series_Title",
        description="The title of the movie",
        type="str",
    ),
    AttributeInfo(
        name="Genre",
        description="The genre of the movie",
        type="string",
    ),
    AttributeInfo(
        name="IMDB_Rating", description="A 1-10 rating for the movie", type="float"
    ),
]
document_content_description = "Brief summary of a movie"
llm = ChatOpenAI(temperature=0)
retriever = SelfQueryRetriever.from_llm(
    llm, vectorstore, document_content_description, metadata_field_info, verbose=True
)

In [10]:

from langchain_core.runnables import RunnablePassthrough

In [11]:

from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

In [12]:

prompt = ChatPromptTemplate.from_template(
    """Answer the user's question based on the below information:

Information:

{info}

Question: {question}"""
)
generator = (prompt | ChatOpenAI() | StrOutputParser()).with_config(
    run_name="generator"
)

In [13]:

chain = (
    RunnablePassthrough.assign(info=(lambda x: x["question"]) | retriever) | generator
)

Run examples

Run examples through the chain. This can either be manually, or using a list of examples, or production traffic

In [14]:

chain.invoke({"question": "what is a horror movie released in early 2000s"})

Out [14]:

'One of the horror movies released in the early 2000s is "The Ring" (2002), directed by Gore Verbinski.'

Annotate

Now, go to LangSmitha and annotate those examples as correct or incorrect

Create Dataset

We can now create a dataset from those runs.

What we will do is find the runs marked as correct, then grab the sub-chains from them. Specifically, the query generator sub chain and the final generation step

In [15]:

from langsmith import Client

client = Client()

In [16]:

runs = list(
    client.list_runs(
        project_name="movie-qa",
        execution_order=1,
        filter="and(eq(feedback_key, 'correctness'), eq(feedback_score, 1))",
    )
)

len(runs)

Out [16]:

14

In [17]:

gen_runs = []
query_runs = []
for r in runs:
    gen_runs.extend(
        list(
            client.list_runs(
                project_name="movie-qa",
                filter="eq(name, 'generator')",
                trace_id=r.trace_id,
            )
        )
    )
    query_runs.extend(
        list(
            client.list_runs(
                project_name="movie-qa",
                filter="eq(name, 'query_constructor')",
                trace_id=r.trace_id,
            )
        )
    )

In [21]:

runs[0].inputs

Out [21]:

{'question': 'what is a high school comedy released in early 2000s'}

In [20]:

runs[0].outputs

Out [20]:

{'output': 'One high school comedy released in the early 2000s is "Mean Girls" starring Lindsay Lohan, Rachel McAdams, and Tina Fey.'}

In [22]:

query_runs[0].inputs

Out [22]:

{'query': 'what is a high school comedy released in early 2000s'}

In [23]:

query_runs[0].outputs

Out [23]:

{'output': {'query': 'high school comedy',
  'filter': {'operator': 'and',
   'arguments': [{'comparator': 'eq', 'attribute': 'Genre', 'value': 'comedy'},
    {'operator': 'and',
     'arguments': [{'comparator': 'gte',
       'attribute': 'Released_Year',
       'value': 2000},
      {'comparator': 'lt', 'attribute': 'Released_Year', 'value': 2010}]}]}}}

In [24]:

gen_runs[0].inputs

Out [24]:

{'question': 'what is a high school comedy released in early 2000s',
 'info': []}

In [25]:

gen_runs[0].outputs

Out [25]:

{'output': 'One high school comedy released in the early 2000s is "Mean Girls" starring Lindsay Lohan, Rachel McAdams, and Tina Fey.'}

Create datasets

We can now create datasets for the query generation and final generation step. We do this so that (1) we can inspect the datapoints, (2) we can edit them if needed, (3) we can add to them over time

In [15]:

client.create_dataset("movie-query_constructor")

inputs = [r.inputs for r in query_runs]
outputs = [r.outputs for r in query_runs]

client.create_examples(
    inputs=inputs, outputs=outputs, dataset_name="movie-query_constructor"
)

In [16]:

client.create_dataset("movie-generator")

inputs = [r.inputs for r in gen_runs]
outputs = [r.outputs for r in gen_runs]

client.create_examples(inputs=inputs, outputs=outputs, dataset_name="movie-generator")

Use as few shot examples

We can now pull down a dataset and use them as few shot examples in a future chain

In [26]:

examples = list(client.list_examples(dataset_name="movie-query_constructor"))

In [27]:

import json


def filter_to_string(_filter):
    if "operator" in _filter:
        args = [filter_to_string(f) for f in _filter["arguments"]]
        return f"{_filter['operator']}({','.join(args)})"
    else:
        comparator = _filter["comparator"]
        attribute = json.dumps(_filter["attribute"])
        value = json.dumps(_filter["value"])
        return f"{comparator}({attribute}, {value})"

In [28]:

model_examples = []

for e in examples:
    if "filter" in e.outputs["output"]:
        string_filter = filter_to_string(e.outputs["output"]["filter"])
    else:
        string_filter = "NO_FILTER"
    model_examples.append(
        (
            e.inputs["query"],
            {"query": e.outputs["output"]["query"], "filter": string_filter},
        )
    )

In [29]:

retriever1 = SelfQueryRetriever.from_llm(
    llm,
    vectorstore,
    document_content_description,
    metadata_field_info,
    verbose=True,
    chain_kwargs={"examples": model_examples},
)

In [30]:

chain1 = (
    RunnablePassthrough.assign(info=(lambda x: x["question"]) | retriever1) | generator
)

In [31]:

chain1.invoke(
    {"question": "what are good action movies made before 2000 but after 1997?"}
)

Out [31]:

'1. "Saving Private Ryan" (1998) - Directed by Steven Spielberg, this war film follows a group of soldiers during World War II as they search for a missing paratrooper.\n\n2. "The Matrix" (1999) - Directed by the Wachowskis, this science fiction action film follows a computer hacker who discovers the truth about the reality he lives in.\n\n3. "Lethal Weapon 4" (1998) - Directed by Richard Donner, this action-comedy film follows two mismatched detectives as they investigate a Chinese immigrant smuggling ring.\n\n4. "The Fifth Element" (1997) - Directed by Luc Besson, this science fiction action film follows a cab driver who must protect a mysterious woman who holds the key to saving the world.\n\n5. "The Rock" (1996) - Directed by Michael Bay, this action thriller follows a group of rogue military men who take over Alcatraz and threaten to launch missiles at San Francisco.'

In [ ]: