File size: 1,249 Bytes
bf93486 |
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 |
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}"
|