diff --git a/agenta-web/src/lib/helpers/openapi_parser.ts b/agenta-web/src/lib/helpers/openapi_parser.ts
index 5e87360d5..57e39bb26 100644
--- a/agenta-web/src/lib/helpers/openapi_parser.ts
+++ b/agenta-web/src/lib/helpers/openapi_parser.ts
@@ -21,11 +21,9 @@ const getBodySchemaName = (schema: GenericObject): string => {
 
 export const detectChatVariantFromOpenAISchema = (schema: GenericObject) => {
     const bodySchemaName = getBodySchemaName(schema)
-    const bodyProperties = schema.components.schemas[bodySchemaName].properties
-    const propertiesWithMessages = Object.keys(bodyProperties).filter((property) => {
-        return bodyProperties[property]["x-parameter"] === "messages"
-    })
-    return propertiesWithMessages[0] === "messages"
+    return (
+        schema.components.schemas[bodySchemaName].properties?.inputs?.["x-parameter"] === "messages"
+    )
 }
 
 export const openAISchemaToParameters = (schema: GenericObject): Parameter[] => {
diff --git a/examples/startup_technical_ideas/app.py b/examples/startup_technical_ideas/app.py
index 8838e5185..6b4241003 100644
--- a/examples/startup_technical_ideas/app.py
+++ b/examples/startup_technical_ideas/app.py
@@ -23,8 +23,8 @@
 
 
 @ag.entrypoint
-def chat(messages: MessagesInput = MessagesInput()) -> str:
-    messages = [{"role": "system", "content": ag.config.prompt_system}] + messages
+def chat(inputs: MessagesInput = MessagesInput()) -> str:
+    messages = [{"role": "system", "content": ag.config.prompt_system}] + inputs
     max_tokens = ag.config.max_tokens if ag.config.max_tokens != -1 else None
     chat_completion = client.chat.completions.create(
         model=ag.config.model,