codeflash 0.3.0__tar.gz → 0.3.2__tar.gz

{codeflash-0.3.0 → codeflash-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codeflash
-Version: 0.3.0
+Version: 0.3.2
 Summary: A Code Performance Optimization Library
 Home-page: https://codeflash.ai
 Author: CodeFlash Inc.

{codeflash-0.3.0 → codeflash-0.3.2}/codeflash/api/aiservice.py

@@ -1,12 +1,12 @@
 import logging
 import os
+import requests
+from pydantic import RootModel
 from typing import Any, Dict, List, Tuple, Optional
 
-import requests
 from codeflash.analytics.posthog import ph
 from codeflash.code_utils.env_utils import get_codeflash_api_key
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
-from pydantic import RootModel
 
 if os.environ.get("AIS_SERVER", default="prod").lower() == "local":
     AI_SERVICE_BASE_URL = "http://localhost:8000/"
@@ -68,7 +68,10 @@ def optimize_python_code(
         optimizations = response.json()["optimizations"]
         return [(opt["source_code"], opt["explanation"]) for opt in optimizations]
     else:
-        error = response.json()["error"]
+        try:
+            error = response.json()["error"]
+        except Exception as e:
+            error = response.text
         logging.error(f"Error generating optimized candidates: {response.status_code} - {error}")
         ph(
             "cli-optimize-error-response",
@@ -129,10 +132,18 @@ def generate_regression_tests(
         response_json = response.json()
         return response_json["generated_tests"], response_json["instrumented_tests"]
     else:
-        error = response.json()["error"]
-        logging.error(f"Error generating tests: {response.status_code} - {error}")
-        ph(
-            "cli-testgen-error-response",
-            {"response_status_code": response.status_code, "error": error},
-        )
-        return None
+        try:
+            error = response.json()["error"]
+            logging.error(f"Error generating tests: {response.status_code} - {error}")
+            ph(
+                "cli-testgen-error-response",
+                {"response_status_code": response.status_code, "error": error},
+            )
+            return None
+        except Exception as e:
+            logging.error(f"Error generating tests: {response.status_code} - {response.text}")
+            ph(
+                "cli-testgen-error-response",
+                {"response_status_code": response.status_code, "error": response.text},
+            )
+            return None

{codeflash-0.3.0 → codeflash-0.3.2}/codeflash/version.py

@@ -1,3 +1,3 @@
 # These version placeholders will be replaced by poetry-dynamic-versioning during `poetry build`.
-__version__ = "0.3.0"
-__version_tuple__ = (0, 3, 0)
+__version__ = "0.3.2"
+__version_tuple__ = (0, 3, 2)

{codeflash-0.3.0 → codeflash-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codeflash"
-version = "0.3.0" # Determined by poetry-dynamic-versioning during `poetry build`
+version = "0.3.2" # Determined by poetry-dynamic-versioning during `poetry build`
 description = "A Code Performance Optimization Library"
 authors = ["CodeFlash Inc. <contact@codeflash.ai>"]
 homepage = "https://codeflash.ai"