abacusai.llm_execution_result

Classes

`ExecuteFeatureGroupOperation`	The result of executing a SQL query
`LlmExecutionPreview`	Preview of executing queries using LLM.
`AbstractApiClass`
`LlmExecutionResult`	Results of executing queries using LLM.

Module Contents

class abacusai.llm_execution_result.ExecuteFeatureGroupOperation(client, featureGroupOperationRunId=None, status=None, error=None, query=None)

Bases: abacusai.return_class.AbstractApiClass

The result of executing a SQL query

Parameters:

client (ApiClient) – An authenticated API Client instance
featureGroupOperationRunId (str) – The run id of the operation
status (str) – The status of the operation
error (str) – The error message if the operation failed
query (str) – The SQL query of the operation

__repr__(): Return repr(self).

to_dict()

Get a dict representation of the parameters in this class

Returns:: The dict value representation of the class parameters
Return type:: dict

wait_for_results(timeout=3600, delay=2)

A waiting call until query is executed.

Parameters:

timeout (int) – The waiting time given to the call to finish, if it doesn’t finish by the allocated time, the call is said to be timed out.
delay (int) – Polling interval for checking timeout.

wait_for_execution(timeout=3600, delay=2)

A waiting call until query is executed.

Parameters:

timeout (int) – The waiting time given to the call to finish, if it doesn’t finish by the allocated time, the call is said to be timed out.
delay (int) – Polling interval for checking timeout.

get_status()

Gets the status of the query execution

Returns:: A string describing the status of a query execution (pending, complete, etc.).
Return type:: str

refresh()

Calls describe and refreshes the current object’s fields

Returns:: The current object
Return type:: DatasetVersion

describe()

Gets the description of the query execution

Returns:: A ExecuteFeatureGroupOperation object describing the query execution.
Return type:: ExecuteFeatureGroupOperation

_download_avro_file(file_part, tmp_dir, part_index)

load_as_pandas(max_workers=10)

Loads the result data into a pandas dataframe

Parameters:: max_workers (int) – The number of threads.
Returns:: A pandas dataframe displaying the data from execution.
Return type:: DataFrame

class abacusai.llm_execution_result.LlmExecutionPreview(client, error=None, sql=None)

Bases: abacusai.return_class.AbstractApiClass

Preview of executing queries using LLM.

Parameters:

client (ApiClient) – An authenticated API Client instance
error (str) – The error message if the preview failed.
sql (str) – Preview of SQL query generated by LLM.

__repr__(): Return repr(self).

to_dict()

Get a dict representation of the parameters in this class

Returns:: The dict value representation of the class parameters
Return type:: dict

class abacusai.llm_execution_result.AbstractApiClass(client, id)

__eq__(other): Return self==value.

_get_attribute_as_dict(attribute)

class abacusai.llm_execution_result.LlmExecutionResult(client, status=None, error=None, execution={}, preview={})

Bases: abacusai.return_class.AbstractApiClass

Results of executing queries using LLM.

Parameters:

client (ApiClient) – An authenticated API Client instance
status (str) – The status of the execution.
error (str) – The error message if the execution failed.
execution (ExecuteFeatureGroupOperation) – Information on execution of the query.
preview (LlmExecutionPreview) – Preview of executing queries using LLM.

__repr__(): Return repr(self).

to_dict()

Get a dict representation of the parameters in this class

Returns:: The dict value representation of the class parameters
Return type:: dict