lint: integrations service ruff format

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Nov 29, 2024 · c095f09 · c095f09
1 parent 73b9321
commit c095f09
Show file tree

Hide file tree

Showing 9 changed files with 12 additions and 16 deletions.
diff --git a/agents-api/gunicorn_conf.py b/agents-api/gunicorn_conf.py
@@ -4,7 +4,7 @@
 debug = os.getenv("AGENTS_API_DEBUG", "false").lower() == "true"
 
 # Gunicorn config variables
-workers = multiprocessing.cpu_count() - 1 if not debug else 1
+workers = (multiprocessing.cpu_count() - 1) if not debug else 1
 worker_class = "uvicorn.workers.UvicornWorker"
 bind = "0.0.0.0:8080"
 keepalive = 120

diff --git a/integrations-service/gunicorn_conf.py b/integrations-service/gunicorn_conf.py
@@ -3,9 +3,14 @@
 
 TESTING = os.getenv("TESTING", "false").lower() == "true"
 DEBUG = os.getenv("DEBUG", "false").lower() == "true"
+AGENTS_API_DEBUG = os.getenv("AGENTS_API_DEBUG", "false").lower() == "true"
 
 # Gunicorn config variables
-workers = multiprocessing.cpu_count() // 2 if not (TESTING or DEBUG) else 1
+workers = (
+    (multiprocessing.cpu_count() // 2)
+    if not (TESTING or AGENTS_API_DEBUG or DEBUG)
+    else 1
+)
 worker_class = "uvicorn.workers.UvicornWorker"
 bind = "0.0.0.0:8000"
 keepalive = 120

diff --git a/integrations-service/integrations/autogen/Tasks.py b/integrations-service/integrations/autogen/Tasks.py
@@ -9,7 +9,6 @@
 from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, StrictBool
 
 from .Chat import ChatSettings
-from .Common import JinjaTemplate
 from .Tools import (
     ChosenBash20241022,
     ChosenComputer20241022,

diff --git a/integrations-service/integrations/autogen/Tools.py b/integrations-service/integrations/autogen/Tools.py
@@ -12,7 +12,6 @@
     BaseModel,
     ConfigDict,
     Field,
-    RootModel,
     StrictBool,
 )
 

diff --git a/integrations-service/integrations/utils/integrations/arxiv.py b/integrations-service/integrations/utils/integrations/arxiv.py
@@ -1,5 +1,4 @@
 import base64
-import os
 import tempfile
 
 import arxiv

diff --git a/integrations-service/integrations/utils/integrations/browserbase.py b/integrations-service/integrations/utils/integrations/browserbase.py
@@ -22,10 +22,10 @@
     BrowserbaseListSessionsArguments,
     BrowserbaseSetup,
 )
-from ...env import (  # Import env to access environment variables  # Import env to access environment variables
+from ...env import (
     browserbase_api_key,
     browserbase_project_id,
-)
+)  # Import env to access environment variables
 from ...models import (
     BrowserbaseCompleteSessionOutput,
     BrowserbaseCreateSessionOutput,

diff --git a/integrations-service/integrations/utils/integrations/llama_parse.py b/integrations-service/integrations/utils/integrations/llama_parse.py
@@ -6,9 +6,7 @@
 from tenacity import retry, stop_after_attempt, wait_exponential
 
 from ...autogen.Tools import LlamaParseFetchArguments, LlamaParseSetup
-from ...env import (
-    llama_api_key,  # Import env to access environment variables  # Import env to access environment variables
-)
+from ...env import llama_api_key  # Import env to access environment variables
 from ...models import LlamaParseFetchOutput
 
 

diff --git a/integrations-service/integrations/utils/integrations/remote_browser.py b/integrations-service/integrations/utils/integrations/remote_browser.py
@@ -1,6 +1,6 @@
 import base64
 import json
-from functools import partial, wraps
+from functools import partial
 from io import BytesIO
 from pathlib import Path
 from typing import Any

diff --git a/integrations-service/integrations/utils/integrations/spider.py b/integrations-service/integrations/utils/integrations/spider.py
@@ -1,15 +1,11 @@
 import asyncio
-from functools import lru_cache
-from typing import Any, Dict
 
 from beartype import beartype
 from langchain_community.document_loaders import SpiderLoader
 from tenacity import retry, stop_after_attempt, wait_exponential
 
 from ...autogen.Tools import SpiderFetchArguments, SpiderSetup
-from ...env import (
-    spider_api_key,  # Import env to access environment variables  # Import env to access environment variables
-)
+from ...env import spider_api_key  # Import env to access environment variables
 from ...models import SpiderFetchOutput