Nina
Adding explanation & background & utils file
bf93486
raw
history blame
1.25 kB
import numpy as np
import openai
import os
def is_climate_change_related(sentence: str, classifier) -> bool:
"""_summary_
Args:
sentence (str): _description_
classifier (_type_): _description_
Returns:
bool: _description_
"""
results = classifier(
sequences=sentence,
candidate_labels=["climate change related", "non climate change related"],
)
print(f" ## Result from is climate change related {results}")
return results["labels"][np.argmax(results["scores"])] == "climate change related"
def make_pairs(lst):
"""From a list of even lenght, make tupple pairs
Args:
lst (list): _description_
Returns:
list: _description_
"""
return [(lst[i], lst[i + 1]) for i in range(0, len(lst), 2)]
def set_openai_api_key(text):
"""Set the api key and return chain.If no api_key, then None is returned.
To do : add raise error & Warning message
Args:
text (str): openai api key
Returns:
str: Result of connection
"""
openai.api_key = os.environ["api_key"]
if text.startswith("sk-") and len(text) > 10:
openai.api_key = text
return f"You're all set: this is your api key: {openai.api_key}"