From dc0218d34c5ebeae57dca5212a3ebee3531ec07e Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 29 Jun 2024 19:24:00 -0700
Subject: [PATCH] build(pyproject.toml): remove ijson dep. add jsonschema dep.

---
 litellm/llms/vertex_httpx.py | 1 -
 pyproject.toml               | 2 +-
 requirements.txt             | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/litellm/llms/vertex_httpx.py b/litellm/llms/vertex_httpx.py
index 91a2b0276..9e361d3cc 100644
--- a/litellm/llms/vertex_httpx.py
+++ b/litellm/llms/vertex_httpx.py
@@ -12,7 +12,6 @@ from functools import partial
 from typing import Any, Callable, Dict, List, Literal, Optional, Tuple, Union
 
 import httpx  # type: ignore
-import ijson
 import requests  # type: ignore
 
 import litellm
diff --git a/pyproject.toml b/pyproject.toml
index 3926ba0bc..578c50cc1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,7 @@ jinja2 = "^3.1.2"
 aiohttp = "*"
 requests = "^2.31.0"
 pydantic = "^2.0.0"
-ijson = "*"
+jsonschema = "^4.22.0"
 
 uvicorn = {version = "^0.22.0", optional = true}
 gunicorn = {version = "^22.0.0", optional = true}
diff --git a/requirements.txt b/requirements.txt
index 00d3802da..e71ab450b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -46,5 +46,5 @@ aiohttp==3.9.0 # for network calls
 aioboto3==12.3.0 # for async sagemaker calls
 tenacity==8.2.3  # for retrying requests, when litellm.num_retries set
 pydantic==2.7.1 # proxy + openai req.
-ijson==3.2.3 # for google ai studio streaming
+jsonschema==4.22.0 # validating json schema
 ####
\ No newline at end of file