fix: add missing dependencies and correct badge URLs

README.md

@@ -1,7 +1,7 @@
 # OpenGradient Python SDK
 
-[![Checks](https://github.com/OpenGradient/sdk/actions/workflows/test.yml/badge.svg)](https://github.com/OpenGradient/sdk/actions/workflows/test.yml)
-[![E2E Tests](https://github.com/OpenGradient/sdk/actions/workflows/e2e.yml/badge.svg)](https://github.com/OpenGradient/sdk/actions/workflows/e2e.yml)
+[![Checks](https://github.com/OpenGradient/OpenGradient-SDK/actions/workflows/test.yml/badge.svg)](https://github.com/OpenGradient/OpenGradient-SDK/actions/workflows/test.yml)
+[![E2E Tests](https://github.com/OpenGradient/OpenGradient-SDK/actions/workflows/e2e.yml/badge.svg)](https://github.com/OpenGradient/OpenGradient-SDK/actions/workflows/e2e.yml)
 [![PyPI](https://img.shields.io/pypi/v/opengradient)](https://pypi.org/project/opengradient/)
 
 A Python SDK for decentralized model management and inference services on the OpenGradient platform. The SDK provides programmatic access to distributed AI infrastructure with cryptographic verification capabilities.

pyproject.toml

@@ -27,7 +27,9 @@ dependencies = [
     "langchain>=0.3.7",
     "openai>=1.58.1",
     "pydantic>=2.9.2",
-    "og-x402==0.0.1.dev4"
+    "og-x402==0.0.1.dev4",
+    "requests-toolbelt>=1.0.0",
+    "httpx>=0.27.0"
 ]
 
 [project.optional-dependencies]
@@ -72,7 +74,7 @@ asyncio_mode = "auto"
 
 [tool.ruff]
 line-length = 140
-target-version = "py310"
+target-version = "py311"
 select = ["E", "F", "I", "N"]
 ignore = []
 

src/opengradient/cli.py

@@ -825,14 +825,6 @@ def generate_image(ctx, model: str, prompt: str, output_path: Path, width: int,
         click.echo(f'Generating image with model "{model}"')
         raise NotImplementedError("Image generation is not yet supported.")
 
-        # Save the image
-        with open(output_path, "wb") as f:
-            f.write(image_data)
-
-        click.echo()  # Add a newline for better spacing
-        click.secho("✅ Image generation successful", fg="green", bold=True)
-        click.echo(f"Image saved to: {output_path}")
-
     except Exception as e:
         click.echo(f"Error generating image: {str(e)}")