Skip to content

Commit 758c15e

Browse files
committed
refactor: linting and isort
1 parent 5403a93 commit 758c15e

File tree

13 files changed

+89
-57
lines changed

13 files changed

+89
-57
lines changed

api.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,22 @@
11
#!/usr/bin/env python3
2+
import hashlib
23
import os
34
import time
4-
import hashlib
5-
import numpy as np
5+
import traceback
6+
from typing import Any, Dict, List, Optional
7+
68
import chromadb
9+
import numpy as np
710
from chromadb.config import Settings
8-
from openai import OpenAI
911
from dotenv import load_dotenv
10-
from typing import List, Optional, Dict, Any
11-
from fastapi import FastAPI, HTTPException, Depends, Request, status
12-
from fastapi.responses import JSONResponse
12+
from fastapi import Depends, FastAPI, HTTPException, Request, status
1313
from fastapi.middleware.cors import CORSMiddleware
14+
from fastapi.responses import JSONResponse
15+
from openai import OpenAI
1416
from pydantic import BaseModel, Field
15-
import traceback
1617

1718
# Import custom logger
18-
from logger import get_logger, LoggerMiddleware
19+
from logger import LoggerMiddleware, get_logger
1920

2021
# Load environment variables from .env file
2122
load_dotenv()

cursor_agent.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,14 @@
66
retrieve context from the Ignition RAG system when generating code or answering questions.
77
"""
88

9-
import os
109
import json
11-
import requests
1210
import logging
13-
from typing import List, Dict, Any, Optional, Union
11+
import os
12+
from typing import Any, Dict, List, Optional, Union
13+
14+
import requests
1415
from dotenv import load_dotenv
16+
1517
from logger import get_logger
1618

1719
# Load environment variables

example_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
#!/usr/bin/env python3
2-
import requests
3-
import json
42
import argparse
3+
import json
4+
5+
import requests
56

67

78
def query_rag_api(

indexer.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,20 @@
11
#!/usr/bin/env python3
2-
import os
2+
import hashlib
33
import json
4-
import time
4+
import os
55
import pickle
6-
import tiktoken
7-
import typer
8-
import hashlib
9-
import numpy as np
10-
from typing import List, Dict, Any, Optional
11-
from pathlib import Path
6+
import time
127
from datetime import datetime
8+
from pathlib import Path
9+
from typing import Any, Dict, List, Optional
10+
1311
import chromadb
12+
import numpy as np
13+
import tiktoken
14+
import typer
1415
from chromadb.config import Settings
15-
from openai import OpenAI
1616
from dotenv import load_dotenv
17+
from openai import OpenAI
1718

1819
# Load environment variables from .env file
1920
load_dotenv()

logger.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
#!/usr/bin/env python3
2+
import logging
23
import os
34
import sys
4-
import logging
5-
from logging.handlers import RotatingFileHandler
65
import time
6+
from logging.handlers import RotatingFileHandler
77

88
# Configure logging
99
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()

pyproject.toml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,3 +30,12 @@ markers = [
3030

3131
[tool.ruff]
3232
line-length = 100
33+
34+
[tool.isort]
35+
profile = "black"
36+
line_length = 100
37+
multi_line_output = 3
38+
include_trailing_comma = true
39+
force_grid_wrap = 0
40+
use_parentheses = true
41+
ensure_newline_before_comments = true

tests/e2e/conftest.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
1+
import logging
12
import os
3+
import time
4+
25
import pytest
36
import requests
4-
import time
5-
import logging
67

78
# Configure logging
89
logging.basicConfig(

tests/e2e/test_e2e.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
#!/usr/bin/env python3
2+
import json
23
import os
3-
import requests
4-
import pytest
54
import sys
6-
import json
5+
6+
import pytest
7+
import requests
78

89
# Add parent directory to path
910
sys.path.append(

tests/e2e/test_indexer_e2e.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
11
#!/usr/bin/env python3
2+
import json
23
import os
3-
import requests
4-
import pytest
4+
import shutil
55
import sys
6-
import json
76
import tempfile
8-
import shutil
7+
8+
import pytest
9+
import requests
910

1011
# Add parent directory to path
1112
sys.path.append(

tests/integration/test_integration.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,25 @@
11
#!/usr/bin/env python3
2-
import os
32
import json
4-
import unittest
5-
import tempfile
3+
import os
64
import shutil
7-
import sys
85
import subprocess
6+
import sys
7+
import tempfile
98
import time
10-
import requests
9+
import unittest
1110
from unittest.mock import patch
1211

12+
import requests
13+
1314
# Add parent directory to path
1415
sys.path.append(
1516
os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
1617
)
17-
import indexer
18-
from api import app, MOCK_EMBEDDINGS, mock_embedding
1918
from fastapi.testclient import TestClient
2019

20+
import indexer
21+
from api import MOCK_EMBEDDINGS, app, mock_embedding
22+
2123

2224
class TestIntegration(unittest.TestCase):
2325
"""Integration tests for the Ignition RAG system."""

0 commit comments

Comments
 (0)