Merge pull request #974 from Agenta-AI/gh/update-chat-llm-template

Enhancement - update chat llm template
Agenta-AI · Nov 30, 2023 · 703c774 · 703c774
2 parents eb541a2 + a100fef
commit 703c774
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git 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
@@ -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,