[PYTHON/LANGCHAIN] get_query_constructor_prompt 함수 : FewShotPromptTemplate 객체 만들기

■ get_query_constructor_prompt 함수를 사용해 FewShotPromptTemplate 객체를 만드는 방법을 보여준다.

▶ 예제 코드 (PY)


from langchain.chains.query_constructor.base import AttributeInfo

attributeInfoList = [
    AttributeInfo(
        name        = "genre",
        description = "The genre of the movie. One of ['science fiction', 'comedy', 'drama', 'thriller', 'romance', 'action', 'animated']",
        type        = "string"
    ),
    AttributeInfo(
        name        = "year",
        description = "The year the movie was released",
        type        = "integer"
    ),
    AttributeInfo(
        name        = "director",
        description = "The name of the movie director",
        type        = "string"
    ),
    AttributeInfo(
        name        = "rating",
        description = "A 1-10 rating for the movie",
        type        = "float"
    )
]

documentContentDescription = "Brief summary of a movie"

fewShotPromptTemplate = get_query_constructor_prompt(
    documentContentDescription,
    attributeInfoList
)

from langchain.chains.query_constructor.base import AttributeInfo

attributeInfoList = [

AttributeInfo(

name = "genre",

description = "The genre of the movie. One of ['science fiction', 'comedy', 'drama', 'thriller', 'romance', 'action', 'animated']",

type = "string"

AttributeInfo(

name = "year",

description = "The year the movie was released",

type = "integer"

AttributeInfo(

name = "director",

description = "The name of the movie director",

type = "string"

AttributeInfo(

name = "rating",

description = "A 1-10 rating for the movie",

type = "float"

)

]

documentContentDescription = "Brief summary of a movie"

fewShotPromptTemplate = get_query_constructor_prompt(

documentContentDescription,

attributeInfoList

)

※ pip install langchain 명령을 실행했다.