Merge pull request #72 from JigsawStack/fix/test-cases-and-python-support

fix: missed base_url for ai_scrape url.

Author: Khurdhula-Harshavardhan

jigsawstack/version.py

@@ -1,4 +1,4 @@
-__version__ = "0.3.8"
+__version__ = "0.3.9"
 
 
 def get_version() -> str:

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="jigsawstack",
-    version="0.3.8",
+    version="0.3.9",
     description="JigsawStack - The AI SDK for Python",
     long_description=open("README.md", encoding="utf8").read(),
     long_description_content_type="text/markdown",
@@ -16,7 +16,7 @@
     packages=find_packages(include=["jigsawstack"]),
     install_requires=install_requires,
     zip_safe=False,
-    python_requires=">=3.7",
+    python_requires=">=3.9",
     keywords=["AI", "AI Tooling"],
     setup_requires=["pytest-runner"],
     tests_require=["pytest", "pytest-asyncio"],
@@ -27,10 +27,9 @@
         "License :: OSI Approved :: MIT License",
         "Natural Language :: English",
         "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3.7",
-        "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
     ],
 )

tests/test_ai_scrape.py

@@ -12,8 +12,20 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-jigsaw = jigsawstack.JigsawStack(api_key=os.getenv("JIGSAWSTACK_API_KEY"))
-async_jigsaw = jigsawstack.AsyncJigsawStack(api_key=os.getenv("JIGSAWSTACK_API_KEY"))
+jigsaw = jigsawstack.JigsawStack(
+    base_url=os.getenv("JIGSAWSTACK_BASE_URL") + "/api"
+    if os.getenv("JIGSAWSTACK_BASE_URL")
+    else "https://api.jigsawstack.com",
+    api_key=os.getenv("JIGSAWSTACK_API_KEY"),
+    headers={"x-jigsaw-skip-cache": "true"},
+)
+async_jigsaw = jigsawstack.AsyncJigsawStack(
+    base_url=os.getenv("JIGSAWSTACK_BASE_URL") + "/api"
+    if os.getenv("JIGSAWSTACK_BASE_URL")
+    else "https://api.jigsawstack.com",
+    api_key=os.getenv("JIGSAWSTACK_API_KEY"),
+    headers={"x-jigsaw-skip-cache": "true"},
+)
 
 URL = "https://jigsawstack.com"
 
@@ -70,7 +82,9 @@
         "params": {
             "url": URL,
             "element_prompts": ["user data"],
-            "cookies": [{"name": "session", "value": "test123", "domain": "example.com"}],
+            "cookies": [
+                {"name": "session", "value": "test123", "domain": "example.com"}
+            ],
         },
     },
     {