fix(types): fix union casting for analytics v1

Author: tharropoulos

src/typesense/async_/analytics_rule_v1.py

@@ -78,10 +78,13 @@ async def retrieve(
             RuleSchemaForQueries, RuleSchemaForCounters
         ] = await self.api_call.get(
             self._endpoint_path,
-            entity_type=typing.Union[RuleSchemaForQueries, RuleSchemaForCounters],
+            entity_type=dict,
             as_json=True,
         )
-        return response
+        return typing.cast(
+            typing.Union[RuleSchemaForQueries, RuleSchemaForCounters],
+            response,
+        )
 
     async def delete(self) -> RuleDeleteSchema:
         """

src/typesense/async_/analytics_rules_v1.py

@@ -122,12 +122,12 @@ async def create(
             body=rule,
             params=rule_parameters,
             as_json=True,
-            entity_type=typing.Union[
-                RuleSchemaForCounters,
-                RuleSchemaForQueries,
-            ],
+            entity_type=dict,
+        )
+        return typing.cast(
+            typing.Union[RuleSchemaForCounters, RuleSchemaForQueries],
+            response,
         )
-        return response
 
     @warn_deprecation(  # type: ignore[untyped-decorator]
         "AsyncAnalyticsRulesV1 is deprecated on v30+. Use client.analytics instead.",
@@ -148,10 +148,12 @@ async def upsert(
         Returns:
             Union[RuleSchemaForCounters, RuleCreateSchemaForQueries]: The upserted rule.
         """
-        response = await self.api_call.put(
+        response: typing.Union[
+            RuleSchemaForCounters, RuleCreateSchemaForQueries
+        ] = await self.api_call.put(
             "/".join([self.resource_path, rule_id]),
             body=rule,
-            entity_type=typing.Union[RuleSchemaForQueries, RuleSchemaForCounters],
+            entity_type=dict,
         )
         return typing.cast(
             typing.Union[RuleSchemaForCounters, RuleCreateSchemaForQueries],