diff options
Diffstat (limited to 'caller.py')
| -rw-r--r-- | caller.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/caller.py b/caller.py new file mode 100644 index 0000000..7b8c68f --- /dev/null +++ b/caller.py @@ -0,0 +1,44 @@ +import openai +import re +import yaml + +# Load API key from yaml file +with open('api-key.yaml', 'r') as file: + config = yaml.safe_load(file) + api_key = config['openai']['api_key'] + +# Initialize the OpenAI client +client = openai.OpenAI(api_key=api_key) + +def extract_meeting_details(long_text): + response = client.chat.completions.create( + model="gpt-3.5-turbo", + messages=[ + {"role": "system", "content": "You are an assistant that extracts meeting details from text. You must respond in a specific format that can be parsed as YAML."}, + {"role": "user", "content": f"""Please extract all mentioned meetings from the following text and format the response as YAML: + +{long_text} + +Format the response as: +meetings: + - time: "YYYY-MM-DD HH:MM - HH:MM" + location: "location name" + description: "event description" + title: "meeting title" + - time: "YYYY-MM-DD HH:MM - HH:MM" + location: "location name" + description: "event description" + title: "meeting title" + +If no meetings are found, respond with: +meetings: null"""} + ], + max_tokens=1500, + temperature=0.2 + ) + + try: + result = yaml.safe_load(response.choices[0].message.content) + return result + except yaml.YAMLError: + return {"meetings": None}
\ No newline at end of file |