answerrocket-client 0.2.100__tar.gz → 0.2.102__tar.gz

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: answerrocket-client
-Version: 0.2.100
+Version: 0.2.102
 Summary: Python client for interacting with AnswerRocket's skill API
 Requires-Python: >=3.10.7
 Description-Content-Type: text/markdown

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answer_rocket/__init__.py

@@ -5,7 +5,7 @@ __all__ = [
     'MetaDataFrame'
 ]
 
-__version__ = "0.2.100"
+__version__ = "0.2.102"
 
 from answer_rocket.client import AnswerRocketClient
 from answer_rocket.error import AnswerRocketClientError

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answer_rocket/chat.py

@@ -586,7 +586,9 @@ class Chat:
         op = Operations.query.dataframes_for_entry
         result = self.gql_client.submit(op, get_dataframes_for_entry_args)
 
-        df_dicts = result.chat_entry.answer.report_results[0].gzipped_dataframes_and_metadata
+        df_dicts = [report.gzipped_dataframes_and_metadata for report in result.chat_entry.answer.report_results if report.gzipped_dataframes_and_metadata]
+        # flatten
+        df_dicts = [df for sublist in df_dicts for df in sublist]
 
         def transform_df(df_dict: dict):
             df = pd.read_csv(io.StringIO(df_dict.get("df")))

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answer_rocket/config.py

@@ -627,6 +627,42 @@ class Config:
         except Exception as e:
             return []
 
+    def update_copilot_skill_field(self, copilot_id: UUID, skill_id: UUID, field_name: str, field_value: Any) -> Optional[MaxMutationResponse]:
+        """
+        Update a specific field of a copilot skill.
+
+        Parameters
+        ----------
+        copilot_id : UUID
+            The ID of the copilot.
+        skill_id : UUID
+            The ID of the skill to update.
+        field_name : str
+            The name of the field to update.
+        field_value : Any
+            The new value for the field.
+
+        Returns
+        -------
+        MaxMutationResponse | None
+            The response from the update operation, or None if an error occurs.
+        """
+        try:
+            mutation_args = {
+                'copilotId': str(copilot_id),
+                'copilotSkillId': str(skill_id),
+                'fieldName': field_name,
+                'fieldValue': field_value
+            }
+
+            op = Operations.mutation.update_copilot_skill_field
+
+            result = self._gql_client.submit(op, mutation_args)
+
+            return result.update_copilot_skill_field
+        except Exception as e:
+            return None
+
 
 def _complete_artifact_path(artifact_path: str, resource_base_path=None) -> str:
     """

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answer_rocket/graphql/schema.py

@@ -1244,7 +1244,7 @@ class ModelOverrideResult(sgqlc.types.Type):
 
 class Mutation(sgqlc.types.Type):
     __schema__ = schema
-    __field_names__ = ('create_max_copilot_skill_chat_question', 'update_max_copilot_skill_chat_question', 'delete_max_copilot_skill_chat_question', 'create_max_copilot_question', 'update_max_copilot_question', 'delete_max_copilot_question', 'run_copilot_skill_async', 'clear_copilot_cache', 'create_copilot_test_run', 'run_copilot_test_run', 'cancel_copilot_test_run', 'set_max_agent_workflow', 'import_copilot_skill_from_zip', 'sync_max_skill_repository', 'import_skill_from_repo', 'test_run_copilot_skill', 'get_test_run_output', 'reload_dataset', 'update_database_name', 'update_database_description', 'update_database_llm_description', 'update_database_mermaid_er_diagram', 'update_database_kshot_limit', 'update_dataset_name', 'update_dataset_description', 'update_dataset_date_range', 'update_dataset_data_interval', 'update_dataset_misc_info', 'update_dataset_source', 'update_dataset_query_row_limit', 'update_dataset_use_database_casing', 'update_dataset_kshot_limit', 'create_dataset', 'create_dataset_from_table', 'create_dimension', 'update_dimension', 'delete_dimension', 'create_metric', 'update_metric', 'delete_metric', 'create_database_kshot', 'update_database_kshot_question', 'update_database_kshot_rendered_prompt', 'update_database_kshot_explanation', 'update_database_kshot_sql', 'update_database_kshot_title', 'update_database_kshot_visualization', 'delete_database_kshot', 'create_dataset_kshot', 'update_dataset_kshot_question', 'update_dataset_kshot_rendered_prompt', 'update_dataset_kshot_explanation', 'update_dataset_kshot_sql', 'update_dataset_kshot_title', 'update_dataset_kshot_visualization', 'delete_dataset_kshot', 'update_chat_answer_payload', 'ask_chat_question', 'evaluate_chat_question', 'queue_chat_question', 'cancel_chat_question', 'create_chat_thread', 'add_feedback', 'set_skill_memory', 'share_thread', 'update_loading_message', 'create_chat_artifact', 'delete_chat_artifact', 'generate_pdf_from_layouts', 'send_email')
+    __field_names__ = ('create_max_copilot_skill_chat_question', 'update_max_copilot_skill_chat_question', 'delete_max_copilot_skill_chat_question', 'create_max_copilot_question', 'update_max_copilot_question', 'delete_max_copilot_question', 'run_copilot_skill_async', 'clear_copilot_cache', 'create_copilot_test_run', 'run_copilot_test_run', 'cancel_copilot_test_run', 'update_copilot_skill_field', 'set_max_agent_workflow', 'import_copilot_skill_from_zip', 'sync_max_skill_repository', 'import_skill_from_repo', 'test_run_copilot_skill', 'get_test_run_output', 'reload_dataset', 'update_database_name', 'update_database_description', 'update_database_llm_description', 'update_database_mermaid_er_diagram', 'update_database_kshot_limit', 'update_dataset_name', 'update_dataset_description', 'update_dataset_date_range', 'update_dataset_data_interval', 'update_dataset_misc_info', 'update_dataset_source', 'update_dataset_query_row_limit', 'update_dataset_use_database_casing', 'update_dataset_kshot_limit', 'create_dataset', 'create_dataset_from_table', 'create_dimension', 'update_dimension', 'delete_dimension', 'create_metric', 'update_metric', 'delete_metric', 'create_database_kshot', 'update_database_kshot_question', 'update_database_kshot_rendered_prompt', 'update_database_kshot_explanation', 'update_database_kshot_sql', 'update_database_kshot_title', 'update_database_kshot_visualization', 'delete_database_kshot', 'create_dataset_kshot', 'update_dataset_kshot_question', 'update_dataset_kshot_rendered_prompt', 'update_dataset_kshot_explanation', 'update_dataset_kshot_sql', 'update_dataset_kshot_title', 'update_dataset_kshot_visualization', 'delete_dataset_kshot', 'update_chat_answer_payload', 'ask_chat_question', 'evaluate_chat_question', 'queue_chat_question', 'cancel_chat_question', 'create_chat_thread', 'add_feedback', 'set_skill_memory', 'share_thread', 'update_loading_message', 'create_chat_artifact', 'delete_chat_artifact', 'generate_pdf_from_layouts', 'send_email')
     create_max_copilot_skill_chat_question = sgqlc.types.Field(sgqlc.types.non_null(CreateMaxCopilotSkillChatQuestionResponse), graphql_name='createMaxCopilotSkillChatQuestion', args=sgqlc.types.ArgDict((
         ('copilot_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotId', default=None)),
         ('copilot_skill_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotSkillId', default=None)),
@@ -1311,6 +1311,13 @@ class Mutation(sgqlc.types.Type):
     cancel_copilot_test_run = sgqlc.types.Field(sgqlc.types.non_null(MaxMutationResponse), graphql_name='cancelCopilotTestRun', args=sgqlc.types.ArgDict((
         ('copilot_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotId', default=None)),
         ('copilot_test_run_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotTestRunId', default=None)),
+))
+    )
+    update_copilot_skill_field = sgqlc.types.Field(sgqlc.types.non_null(MaxMutationResponse), graphql_name='updateCopilotSkillField', args=sgqlc.types.ArgDict((
+        ('copilot_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotId', default=None)),
+        ('copilot_skill_id', sgqlc.types.Arg(sgqlc.types.non_null(UUID), graphql_name='copilotSkillId', default=None)),
+        ('field', sgqlc.types.Arg(sgqlc.types.non_null(String), graphql_name='field', default=None)),
+        ('value', sgqlc.types.Arg(sgqlc.types.non_null(JSON), graphql_name='value', default=None)),
 ))
     )
     set_max_agent_workflow = sgqlc.types.Field(sgqlc.types.non_null(MaxMutationResponse), graphql_name='setMaxAgentWorkflow', args=sgqlc.types.ArgDict((

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answer_rocket/graphql/sdk_operations.py

@@ -179,6 +179,14 @@ def mutation_cancel_copilot_test_run():
     return _op
 
 
+def mutation_update_copilot_skill_field():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='UpdateCopilotSkillField', variables=dict(copilotId=sgqlc.types.Arg(sgqlc.types.non_null(_schema.UUID)), copilotSkillId=sgqlc.types.Arg(sgqlc.types.non_null(_schema.UUID)), fieldName=sgqlc.types.Arg(sgqlc.types.non_null(_schema.String)), fieldValue=sgqlc.types.Arg(sgqlc.types.non_null(_schema.JSON))))
+    _op_update_copilot_skill_field = _op.update_copilot_skill_field(copilot_id=sgqlc.types.Variable('copilotId'), copilot_skill_id=sgqlc.types.Variable('copilotSkillId'), field=sgqlc.types.Variable('fieldName'), value=sgqlc.types.Variable('fieldValue'))
+    _op_update_copilot_skill_field.success()
+    _op_update_copilot_skill_field.error()
+    return _op
+
+
 def mutation_update_database_name():
     _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='UpdateDatabaseName', variables=dict(databaseId=sgqlc.types.Arg(sgqlc.types.non_null(_schema.UUID)), name=sgqlc.types.Arg(sgqlc.types.non_null(_schema.String))))
     _op_update_database_name = _op.update_database_name(database_id=sgqlc.types.Variable('databaseId'), name=sgqlc.types.Variable('name'))
@@ -590,6 +598,7 @@ class Mutation:
     set_skill_memory = mutation_set_skill_memory()
     sync_max_skill_repository = mutation_sync_max_skill_repository()
     test_run_copilot_skill = mutation_test_run_copilot_skill()
+    update_copilot_skill_field = mutation_update_copilot_skill_field()
     update_database_description = mutation_update_database_description()
     update_database_kshot_explanation = mutation_update_database_kshot_explanation()
     update_database_kshot_limit = mutation_update_database_kshot_limit()

{answerrocket_client-0.2.100 → answerrocket_client-0.2.102}/answerrocket_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: answerrocket-client
-Version: 0.2.100
+Version: 0.2.102
 Summary: Python client for interacting with AnswerRocket's skill API
 Requires-Python: >=3.10.7
 Description-Content-Type: text/markdown