mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-30 07:48:38 +00:00
480626dc99
…tch]: import models from community ran ```bash git grep -l 'from langchain\.chat_models' | xargs -L 1 sed -i '' "s/from\ langchain\.chat_models/from\ langchain_community.chat_models/g" git grep -l 'from langchain\.llms' | xargs -L 1 sed -i '' "s/from\ langchain\.llms/from\ langchain_community.llms/g" git grep -l 'from langchain\.embeddings' | xargs -L 1 sed -i '' "s/from\ langchain\.embeddings/from\ langchain_community.embeddings/g" git checkout master libs/langchain/tests/unit_tests/llms git checkout master libs/langchain/tests/unit_tests/chat_models git checkout master libs/langchain/tests/unit_tests/embeddings/test_imports.py make format cd libs/langchain; make format cd ../experimental; make format cd ../core; make format ```
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
import json
|
|
from typing import List, Optional
|
|
|
|
from langchain.prompts import ChatPromptTemplate
|
|
from langchain.utils.openai_functions import convert_pydantic_to_openai_function
|
|
from langchain_community.chat_models import ChatOpenAI
|
|
from langchain_core.pydantic_v1 import BaseModel
|
|
|
|
template = """A article will be passed to you. Extract from it all papers that are mentioned by this article.
|
|
|
|
Do not extract the name of the article itself. If no papers are mentioned that's fine - you don't need to extract any! Just return an empty list.
|
|
|
|
Do not make up or guess ANY extra information. Only extract what exactly is in the text.""" # noqa: E501
|
|
|
|
prompt = ChatPromptTemplate.from_messages([("system", template), ("human", "{input}")])
|
|
|
|
|
|
# Function output schema
|
|
class Paper(BaseModel):
|
|
"""Information about papers mentioned."""
|
|
|
|
title: str
|
|
author: Optional[str]
|
|
|
|
|
|
class Info(BaseModel):
|
|
"""Information to extract"""
|
|
|
|
papers: List[Paper]
|
|
|
|
|
|
# Function definition
|
|
model = ChatOpenAI()
|
|
function = [convert_pydantic_to_openai_function(Info)]
|
|
chain = (
|
|
prompt
|
|
| model.bind(functions=function, function_call={"name": "Info"})
|
|
| (
|
|
lambda x: json.loads(x.additional_kwargs["function_call"]["arguments"])[
|
|
"papers"
|
|
]
|
|
)
|
|
)
|
|
|
|
# chain = prompt | model.bind(
|
|
# functions=function, function_call={"name": "Info"}
|
|
# ) | JsonKeyOutputFunctionsParser(key_name="papers")
|