This commit is contained in:
Harrison Chase 2023-01-24 22:10:12 -08:00
parent fd74718447
commit e93f620fc4

View File

@ -2,7 +2,7 @@
"cells": [ "cells": [
{ {
"cell_type": "markdown", "cell_type": "markdown",
"id": "985599b1", "id": "ad719b65",
"metadata": {}, "metadata": {},
"source": [ "source": [
"# Analyze Document\n", "# Analyze Document\n",
@ -15,7 +15,7 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 7, "execution_count": 7,
"id": "73bb17f2", "id": "765d6326",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
@ -32,7 +32,7 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 2, "execution_count": 2,
"id": "31aec990", "id": "3a3d3ebc",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
@ -42,7 +42,7 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 6, "execution_count": 6,
"id": "dcb4bb8c", "id": "97178aad",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
@ -52,7 +52,7 @@
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 8, "execution_count": 8,
"id": "2d15baab", "id": "2e5a7bf7",
"metadata": {}, "metadata": {},
"outputs": [ "outputs": [
{ {
@ -70,10 +70,69 @@
"summarize_document_chain.run(state_of_the_union)" "summarize_document_chain.run(state_of_the_union)"
] ]
}, },
{
"cell_type": "markdown",
"id": "35739404",
"metadata": {},
"source": [
"Let's take a look at this using a question answering chain."
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "8b9b7705",
"metadata": {},
"outputs": [],
"source": [
"from langchain.chains.question_answering import load_qa_chain"
]
},
{
"cell_type": "code",
"execution_count": 10,
"id": "60c309a8",
"metadata": {},
"outputs": [],
"source": [
"qa_chain = load_qa_chain(llm, chain_type=\"map_reduce\")"
]
},
{
"cell_type": "code",
"execution_count": 11,
"id": "ba1fc940",
"metadata": {},
"outputs": [],
"source": [
"qa_document_chain = AnalyzeDocumentChain(combine_docs_chain=qa_chain)"
]
},
{
"cell_type": "code",
"execution_count": 13,
"id": "9aa1fbde",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"' The president thanked Justice Breyer for his service.'"
]
},
"execution_count": 13,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"qa_document_chain.run(input_document=state_of_the_union, question=\"what did the president say about justice breyer?\")"
]
},
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": null,
"id": "a81b2c8e", "id": "7eb02f1e",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [] "source": []