From 42f44417e262ef4dbd7a27f79ef4ec8fda9fd270 Mon Sep 17 00:00:00 2001 From: Matt Haidet Date: Fri, 24 Oct 2025 10:09:43 -0400 Subject: [PATCH] Update schema with model_selection for chat/research with prompt calls --- answer_rocket/graphql/schema.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/answer_rocket/graphql/schema.py b/answer_rocket/graphql/schema.py index f47a95d..f2daae2 100644 --- a/answer_rocket/graphql/schema.py +++ b/answer_rocket/graphql/schema.py @@ -1781,6 +1781,7 @@ class Query(sgqlc.types.Type): ('prompt_name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='promptName', default=None)), ('prompt_variables', sgqlc.types.Arg(JSON, graphql_name='promptVariables', default=None)), ('messages', sgqlc.types.Arg(sgqlc.types.list_of(sgqlc.types.non_null(LlmChatMessage)), graphql_name='messages', default=None)), + ('model_selection', sgqlc.types.Arg(LlmModelSelection, graphql_name='modelSelection', default=None)), ('llm_meta', sgqlc.types.Arg(LlmMeta, graphql_name='llmMeta', default=None)), ('functions', sgqlc.types.Arg(sgqlc.types.list_of(sgqlc.types.non_null(LlmFunction)), graphql_name='functions', default=None)), )) @@ -1789,6 +1790,7 @@ class Query(sgqlc.types.Type): ('prompt_name', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='promptName', default=None)), ('prompt_variables', sgqlc.types.Arg(JSON, graphql_name='promptVariables', default=None)), ('messages', sgqlc.types.Arg(sgqlc.types.list_of(sgqlc.types.non_null(LlmChatMessage)), graphql_name='messages', default=None)), + ('model_selection', sgqlc.types.Arg(LlmModelSelection, graphql_name='modelSelection', default=None)), ('llm_meta', sgqlc.types.Arg(LlmMeta, graphql_name='llmMeta', default=None)), ('functions', sgqlc.types.Arg(sgqlc.types.list_of(sgqlc.types.non_null(LlmFunction)), graphql_name='functions', default=None)), ))