diff --git a/jigsawstack/version.py b/jigsawstack/version.py index 2819f1c..b8d2d3c 100644 --- a/jigsawstack/version.py +++ b/jigsawstack/version.py @@ -1,4 +1,4 @@ -__version__ = "0.3.8" +__version__ = "0.3.9" def get_version() -> str: diff --git a/setup.py b/setup.py index 4524e2c..8a9a443 100644 --- a/setup.py +++ b/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", ], ) diff --git a/tests/test_ai_scrape.py b/tests/test_ai_scrape.py index 4c30b33..1f53453 100644 --- a/tests/test_ai_scrape.py +++ b/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"} + ], }, }, {