File size: 4,017 Bytes
481f3b1 088e816 481f3b1 48e003d 6b43c86 0c4d82b 481f3b1 48e003d 481f3b1 6b43c86 0c4d82b 481f3b1 088e816 481f3b1 088e816 481f3b1 0c4d82b 481f3b1 088e816 481f3b1 088e816 481f3b1 088e816 481f3b1 088e816 49acaf1 6b43c86 481f3b1 088e816 481f3b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
from langchain_core.pydantic_v1 import BaseModel, Field
from typing import List
from typing import Literal
from langchain.prompts import ChatPromptTemplate
from langchain_core.utils.function_calling import convert_to_openai_function
from langchain.output_parsers.openai_functions import JsonOutputFunctionsParser
class IntentCategorizer(BaseModel):
"""Analyzing the user message input"""
language: str = Field(
description="Find the language of the message input in full words (ex: French, English, Spanish, ...), defaults to English",
default="English",
)
intent: str = Field(
enum=[
"ai_impact",
# "geo_info",
# "esg",
"search",
"chitchat",
],
description="""
Categorize the user input in one of the following category
Any question
Examples:
- ai_impact = Environmental impacts of AI: "What are the environmental impacts of AI", "How does AI affect the environment"
- search = Searching for any quesiton about climate change, energy, biodiversity, nature, and everything we can find the IPCC or IPBES reports or scientific papers,
- chitchat = Any general question that is not related to the environment or climate change or just conversational, or if you don't think searching the IPCC or IPBES reports would be relevant
""",
# - geo_info = Geolocated info about climate change: Any question where the user wants to know localized impacts of climate change, eg: "What will be the temperature in Marseille in 2050"
# - esg = Any question about the ESG regulation, frameworks and standards like the CSRD, TCFD, SASB, GRI, CDP, etc.
)
def make_intent_categorization_chain(llm):
openai_functions = [convert_to_openai_function(IntentCategorizer)]
llm_with_functions = llm.bind(functions = openai_functions,function_call={"name":"IntentCategorizer"})
prompt = ChatPromptTemplate.from_messages([
("system", "You are a helpful assistant, you will analyze, translate and categorize the user input message using the function provided. Categorize the user input as ai ONLY if it is related to Artificial Intelligence, search if it is related to the environment, climate change, energy, biodiversity, nature, etc. and chitchat if it is just general conversation."),
("user", "input: {input}")
])
chain = prompt | llm_with_functions | JsonOutputFunctionsParser()
return chain
def make_intent_categorization_node(llm):
categorization_chain = make_intent_categorization_chain(llm)
def categorize_message(state):
print("---- Categorize_message ----")
output = categorization_chain.invoke({"input": state["user_input"]})
print(f"\n\nOutput intent categorization: {output}\n")
if "language" not in output: output["language"] = "English"
output["query"] = state["user_input"]
return output
return categorize_message
# SAMPLE_QUESTIONS = [
# "Est-ce que l'IA a un impact sur l'environnement ?",
# "Que dit le GIEC sur l'impact de l'IA",
# "Qui sont les membres du GIEC",
# "What is the impact of El Nino ?",
# "Yo",
# "Hello ça va bien ?",
# "Par qui as tu été créé ?",
# "What role do cloud formations play in modulating the Earth's radiative balance, and how are they represented in current climate models?",
# "Which industries have the highest GHG emissions?",
# "What are invasive alien species and how do they threaten biodiversity and ecosystems?",
# "Are human activities causing global warming?",
# "What is the motivation behind mining the deep seabed?",
# "Tu peux m'écrire un poème sur le changement climatique ?",
# "Tu peux m'écrire un poème sur les bonbons ?",
# "What will be the temperature in 2100 in Strasbourg?",
# "C'est quoi le lien entre biodiversity and changement climatique ?",
# ] |