Add functions to invoke Bedrock and OpenAI models

utils.py

@@ -1,14 +1,36 @@
 import os
+import json
 from io import BytesIO
 
 import boto3
 import requests
+from openai import OpenAI
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 BEDROCK_MODEL_ID = "anthropic.claude-3-5-sonnet-20240620-v1:0"
 OPENAI_MODEL_ID = "o4-mini"
+OPENAI_MODEL_IMAGE_MODEL_ID = "gpt-4.1-mini"
 
 bedrock_runtime = boto3.client("bedrock-runtime", region_name=os.getenv("AWS_REGION"))
+openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+
+def invoke_bedrock_model(messages: list[dict]) -> str:
+    """Invokes a Bedrock model with the provided messages."""
+    response = bedrock_runtime.invoke_model(
+        modelId=BEDROCK_MODEL_ID,
+        body=json.dumps({"anthropic_version": "bedrock-2023-05-31", "max_tokens": 4096, "messages": messages}),
+    )["body"].read()
+    return json.loads(response)["content"][0]["text"]
+
+
+def invoke_openai_model(messages: list[dict]) -> str:
+    """Invokes an OpenAI model with the provided messages."""
+    response = openai_client.responses.create(
+        model=OPENAI_MODEL_IMAGE_MODEL_ID,
+        input=messages,
+    )
+    return response.output_text
 
 
 def get_file(task_id: str) -> BytesIO: