finish making tests work

generated/docs/ImageQuery.md

@@ -11,7 +11,7 @@ Name | Type | Description | Notes
 **query** | **str** | A question about the image. | [readonly] 
 **detector_id** | **str** | Which detector was used on this image query? | [readonly] 
 **result_type** | **bool, date, datetime, dict, float, int, list, str, none_type** | What type of result are we returning? | [readonly] 
-**result** | **bool, date, datetime, dict, float, int, list, str, none_type** |  | [readonly] 
+**result** | **bool, date, datetime, dict, float, int, list, str, none_type** |  | [optional] [readonly] 
 **any string name** | **bool, date, datetime, dict, float, int, list, str, none_type** | any string name can be used but the value must be the correct type | [optional]
 
 [[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)

generated/model.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  public-api.yaml
-#   timestamp: 2023-10-16T22:49:17+00:00
+#   timestamp: 2023-10-16T23:29:00+00:00
 
 from __future__ import annotations
 
@@ -69,7 +69,7 @@ class ImageQuery(BaseModel):
     query: str = Field(..., description="A question about the image.")
     detector_id: str = Field(..., description="Which detector was used on this image query?")
     result_type: ResultTypeEnum = Field(..., description="What type of result are we returning?")
-    result: ClassificationResult
+    result: Optional[ClassificationResult] = None
 
 
 class PaginatedDetectorList(BaseModel):

generated/openapi_client/model/image_query.py

@@ -168,9 +168,7 @@ def discriminator():
 
     @classmethod
     @convert_js_args_to_python_args
-    def _from_openapi_data(
-        cls, id, type, created_at, query, detector_id, result_type, result, *args, **kwargs
-    ):  # noqa: E501
+    def _from_openapi_data(cls, id, type, created_at, query, detector_id, result_type, *args, **kwargs):  # noqa: E501
         """ImageQuery - a model defined in OpenAPI
 
         Args:
@@ -180,7 +178,6 @@ def _from_openapi_data(
             query (str): A question about the image.
             detector_id (str): Which detector was used on this image query?
             result_type (bool, date, datetime, dict, float, int, list, str, none_type): What type of result are we returning?
-            result (bool, date, datetime, dict, float, int, list, str, none_type):
 
         Keyword Args:
             _check_type (bool): if True, values for parameters in openapi_types
@@ -213,6 +210,7 @@ def _from_openapi_data(
                                 Animal class but this time we won't travel
                                 through its discriminator because we passed in
                                 _visited_composed_classes = (Animal,)
+            result (bool, date, datetime, dict, float, int, list, str, none_type): [optional]  # noqa: E501
         """
 
         _check_type = kwargs.pop("_check_type", True)
@@ -247,7 +245,6 @@ def _from_openapi_data(
         self.query = query
         self.detector_id = detector_id
         self.result_type = result_type
-        self.result = result
         for var_name, var_value in kwargs.items():
             if (
                 var_name not in self.attribute_map
@@ -306,6 +303,7 @@ def __init__(self, *args, **kwargs):  # noqa: E501
                                 Animal class but this time we won't travel
                                 through its discriminator because we passed in
                                 _visited_composed_classes = (Animal,)
+            result (bool, date, datetime, dict, float, int, list, str, none_type): [optional]  # noqa: E501
         """
 
         _check_type = kwargs.pop("_check_type", True)

spec/public-api.yaml

@@ -344,7 +344,6 @@ components:
         - detector_id
         - id
         - query
-        - result
         - result_type
         - type
       x-internal: true

test/integration/test_groundlight.py

@@ -2,6 +2,7 @@
 # ruff: noqa: F403,F405
 # pylint: disable=wildcard-import,unused-wildcard-import,redefined-outer-name,import-outside-toplevel
 from datetime import datetime
+import time
 from typing import Any
 
 import openapi_client
@@ -265,11 +266,11 @@ def test_submit_image_query_with_want_async_workflow(gl: Groundlight, detector:
     assert _image_query.result is None
 
     # attempting to access fields within the result should raise an exception
-    with pytest.raises(ValueError):
+    with pytest.raises(AttributeError):
         _ = _image_query.result.label  # type: ignore
-    with pytest.raises(ValueError):
+    with pytest.raises(AttributeError):
         _ = _image_query.result.confidence  # type: ignore
-
+    time.sleep(5)
     # you should be able to get a "real" result by retrieving an updated image query object from the server
     _image_query = gl.get_image_query(id=_image_query.id)
     assert _image_query.result is not None
@@ -291,6 +292,8 @@ def test_ask_async_workflow(gl: Groundlight, detector: Detector):
     with pytest.raises(AttributeError):
         _ = _image_query.result.confidence  # type: ignore
 
+    time.sleep(5)
+
     # you should be able to get a "real" result by retrieving an updated image query object from the server
     _image_query = gl.get_image_query(id=_image_query.id)
     assert _image_query.result is not None