Dev

.github/workflows/pytest_and_coverage.yml

@@ -14,6 +14,10 @@ jobs:
       # Logging
       PROMETHEUS_LOGGING_LEVEL: DEBUG
 
+      # General settings
+      PROMETHEUS_ENVIRONMENT: local
+      PROMETHEUS_BACKEND_CORS_ORIGINS: "[\"*\"]"
+
       # Neo4j settings
       PROMETHEUS_NEO4J_URI: bolt://localhost:7687
       PROMETHEUS_NEO4J_USERNAME: neo4j
@@ -48,6 +52,9 @@ jobs:
       # DATABASE settings
       PROMETHEUS_DATABASE_URL: postgresql://postgres:password@localhost:5432/postgres?sslmode=disable
 
+      # JWT settings
+      PROMETHEUS_JWT_SECRET_KEY: your_jwt_secret_key
+
     steps:
       - name: Check out code
         uses: actions/checkout@v4

README.md

@@ -73,10 +73,20 @@ governed by a state machine to ensure code quality through automated reviews, bu
    ```
 
 2. #### Copy the `example.env` file to `.env` and update it with your API keys and other required configurations:
+
    ```bash
    mv example.env .env
    ```
 
+   > You need to provide a secure `JWT_SECRET_KEY` in the `.env` file.
+   > You can generate a strong key by running the following command:
+
+   ```bash
+   python -m prometheus.script.generate_jwt_token
+   ```
+
+   This will print a secure token you can copy and paste into your `.env` file
+
 3. #### Create the working directory to store logs and cloned repositories:
 
    ```bash
@@ -124,7 +134,7 @@ governed by a state machine to ensure code quality through automated reviews, bu
    You can ask Prometheus to analyze and answer a specific issue in your codebase using the `/issue/answer/` API endpoint.
 
    - **Endpoint:** `POST /issue/answer/`
-     - **Request Body:** JSON object matching the `IssueRequest` schema (see [API Documents](http://localhost:9002/docs#/issue/answer_issue_issue_answer__post))
+     - **Request Body:** JSON object matching the `IssueRequest` schema (see [API Documents](http://127.0.0.1:9002/docs#/issue/issue-answer_issue))
      - **Response:** Returns the generated patch, test/build results, and a summary response.
 
 ---

docker-compose.win_mac.yml

@@ -36,6 +36,10 @@ services:
       # Logging
       - PROMETHEUS_LOGGING_LEVEL=${PROMETHEUS_LOGGING_LEVEL}
 
+      # General settings
+      - PROMETHEUS_ENVIRONMENT=${PROMETHEUS_ENVIRONMENT}
+      - PROMETHEUS_BACKEND_CORS_ORIGINS=${PROMETHEUS_BACKEND_CORS_ORIGINS}
+
       # Neo4j settings
       - PROMETHEUS_NEO4J_URI=${PROMETHEUS_NEO4J_URI}
       - PROMETHEUS_NEO4J_USERNAME=${PROMETHEUS_NEO4J_USERNAME}
@@ -64,9 +68,6 @@ services:
       - PROMETHEUS_MAX_OUTPUT_TOKENS=${PROMETHEUS_MAX_OUTPUT_TOKENS}
       - PROMETHEUS_TEMPERATURE=${PROMETHEUS_TEMPERATURE}
 
-      # GitHub settings
-      - PROMETHEUS_GITHUB_ACCESS_TOKEN=${PROMETHEUS_GITHUB_ACCESS_TOKEN}
-
       # Database settings
       - PROMETHEUS_DATABASE_URL=${PROMETHEUS_DATABASE_URL}
     networks:

docker-compose.yml

@@ -57,6 +57,10 @@ services:
       # Logging
       - PROMETHEUS_LOGGING_LEVEL=${PROMETHEUS_LOGGING_LEVEL}
 
+      # General settings
+      - PROMETHEUS_ENVIRONMENT=${PROMETHEUS_ENVIRONMENT}
+      - PROMETHEUS_BACKEND_CORS_ORIGINS=${PROMETHEUS_BACKEND_CORS_ORIGINS}
+
       # Neo4j settings
       - PROMETHEUS_NEO4J_URI=${PROMETHEUS_NEO4J_URI}
       - PROMETHEUS_NEO4J_USERNAME=${PROMETHEUS_NEO4J_USERNAME}
@@ -85,9 +89,6 @@ services:
       - PROMETHEUS_MAX_OUTPUT_TOKENS=${PROMETHEUS_MAX_OUTPUT_TOKENS}
       - PROMETHEUS_TEMPERATURE=${PROMETHEUS_TEMPERATURE}
 
-      # GitHub settings
-      - PROMETHEUS_GITHUB_ACCESS_TOKEN=${PROMETHEUS_GITHUB_ACCESS_TOKEN}
-
       # Database settings
       - PROMETHEUS_DATABASE_URL=${PROMETHEUS_DATABASE_URL}
     volumes:

example.env

@@ -1,6 +1,10 @@
 # Logging
 PROMETHEUS_LOGGING_LEVEL=DEBUG
 
+# General settings
+PROMETHEUS_ENVIRONMENT=local
+PROMETHEUS_BACKEND_CORS_ORIGINS=["*]
+
 # Neo4j settings
 PROMETHEUS_NEO4J_URI=bolt://neo4j:7687
 PROMETHEUS_NEO4J_USERNAME=neo4j
@@ -29,8 +33,8 @@ PROMETHEUS_MAX_INPUT_TOKENS=64000
 PROMETHEUS_TEMPERATURE=0.3
 PROMETHEUS_MAX_OUTPUT_TOKENS=15000
 
-# GitHub settings
-PROMETHEUS_GITHUB_ACCESS_TOKEN=github_access_token
-
 # Database settings
 PROMETHEUS_DATABASE_URL=postgresql://postgres:password@localhost:5432/postgres?sslmode=disable
+
+# JWT settings
+PROMETHEUS_JWT_SECRET_KEY=your_jwt_secret_key

prometheus/app/api/auth.py

@@ -0,0 +1,30 @@
+from fastapi import APIRouter, Request
+
+from prometheus.app.models.requests.auth import LoginRequest
+from prometheus.app.models.response.auth import LoginResponse
+from prometheus.app.models.response.response import Response
+from prometheus.app.services.user_service import UserService
+
+router = APIRouter()
+
+
+@router.post(
+    "/login/",
+    summary="Login to the system",
+    description="Login to the system using username, email, and password. Returns an access token.",
+    response_description="Returns an access token for authenticated requests",
+    response_model=Response[LoginResponse],
+)
+def login(login_request: LoginRequest, request: Request) -> Response[LoginResponse]:
+    """
+    Login to the system using username, email, and password.
+    Returns an access token for authenticated requests.
+    """
+
+    user_service: UserService = request.app.state.service["user_service"]
+    access_token = user_service.login(
+        username=login_request.username,
+        email=login_request.email,
+        password=login_request.password,
+    )
+    return Response(data=LoginResponse(access_token=access_token))

prometheus/app/api/issue.py

@@ -1,105 +1,33 @@
-from typing import Mapping, Optional, Sequence
+from fastapi import APIRouter, Request
 
-from fastapi import APIRouter, HTTPException, Request
-from litellm import Field
-from pydantic import BaseModel
-
-from prometheus.lang_graph.graphs.issue_state import IssueType
+from prometheus.app.models.requests.issue import IssueRequest
+from prometheus.app.models.response.issue import IssueResponse
+from prometheus.app.models.response.response import Response
+from prometheus.exceptions.server_exception import ServerException
 
 router = APIRouter()
 
 
-class IssueRequest(BaseModel):
-    issue_number: int = Field(description="The number of the issue", examples=[42])
-    issue_title: str = Field(
-        description="The title of the issue", examples=["There is a memory leak"]
-    )
-    issue_body: str = Field(
-        description="The description of the issue", examples=["foo/bar.c is causing a memory leak"]
-    )
-    issue_comments: Optional[Sequence[Mapping[str, str]]] = Field(
-        default=None,
-        description="Comments on the issue",
-        examples=[
-            [
-                {"username": "user1", "comment": "I've experienced this issue as well."},
-                {
-                    "username": "user2",
-                    "comment": "A potential fix is to adjust the memory settings.",
-                },
-            ]
-        ],
-    )
-    issue_type: IssueType = Field(
-        default=IssueType.AUTO,
-        description="The type of the issue, set to auto if you do not know",
-        examples=[IssueType.AUTO],
-    )
-    run_build: Optional[bool] = Field(
-        default=False,
-        description="When editing the code, whenver we should run the build to verify the fix",
-        examples=[False],
-    )
-    run_existing_test: Optional[bool] = Field(
-        default=False,
-        description="When editing the code, whenver we should run the existing test to verify the fix",
-        examples=[False],
-    )
-    number_of_candidate_patch: Optional[int] = Field(
-        default=4,
-        description="When the patch is not verfied (through build or test), number of candidate patches we generate to select the best one",
-        examples=[4],
-    )
-    dockerfile_content: Optional[str] = Field(
-        default=None,
-        description="Specify the containerized environment with dockerfile content",
-        examples=["FROM python:3.11\nWORKDIR /app\nCOPY . /app"],
-    )
-    image_name: Optional[str] = Field(
-        default=None,
-        description="Specify the containerized environment with image name that should be pulled from dockerhub",
-        examples=["python:3.11-slim"],
-    )
-    workdir: Optional[str] = Field(
-        default=None,
-        description="If you specified the container environment, you must also specify the workdir",
-        examples=["/app"],
-    )
-    build_commands: Optional[Sequence[str]] = Field(
-        default=None,
-        description="If you specified dockerfile_content and run_build is True, you must also specify the build commands.",
-        examples=[["pip install -r requirements.txt", "python -m build"]],
-    )
-    test_commands: Optional[Sequence[str]] = Field(
-        default=None,
-        description="If you specified dockerfile_content and run_test is True, you must also specify the test commands.",
-        examples=[["pytest ."]],
-    )
-    push_to_remote: Optional[bool] = Field(
-        default=False,
-        description="When editing the code, whenver we should push the changes to a remote branch",
-        examples=[True],
-    )
-
-
 @router.post(
     "/answer/",
     summary="Process and generate a response for an issue",
-    description="Analyzes an issue, generates patches if needed, runs optional builds and tests, and can push changes to a remote branch.",
+    description="Analyzes an issue, generates patches if needed, runs optional builds and tests, and can push changes "
+    "to a remote branch.",
     response_description="Returns the patch, test results, and issue response",
+    response_model=Response[IssueResponse],
 )
-def answer_issue(issue: IssueRequest, request: Request):
-    if not request.app.state.service_coordinator.exists_knowledge_graph():
-        raise HTTPException(
-            status_code=404,
-            detail="A repository is not uploaded, use /repository/ endpoint to upload one",
+def answer_issue(issue: IssueRequest, request: Request) -> Response[IssueResponse]:
+    if not request.app.state.service["knowledge_graph_service"].exists():
+        raise ServerException(
+            code=404,
+            message="A repository is not uploaded, use /repository/ endpoint to upload one",
         )
 
     if issue.dockerfile_content or issue.image_name:
         if issue.workdir is None:
-            raise HTTPException(
-                status_code=400,
-                detail="workdir must be provided for user defined environment",
+            raise ServerException(
+                code=400,
+                message="workdir must be provided for user defined environment",
             )
 
     (
@@ -109,7 +37,7 @@ def answer_issue(issue: IssueRequest, request: Request):
         passed_build,
         passed_existing_test,
         issue_response,
-    ) = request.app.state.service_coordinator.answer_issue(
+    ) = request.app.state.service["issue_service"].answer_issue(
         issue_number=issue.issue_number,
         issue_title=issue.issue_title,
         issue_body=issue.issue_body,
@@ -125,11 +53,13 @@ def answer_issue(issue: IssueRequest, request: Request):
         test_commands=issue.test_commands,
         push_to_remote=issue.push_to_remote,
     )
-    return {
-        "patch": patch,
-        "passed_reproducing_test": passed_reproducing_test,
-        "passed_build": passed_build,
-        "passed_existing_test": passed_existing_test,
-        "issue_response": issue_response,
-        "remote_branch_name": remote_branch_name,
-    }
+    return Response(
+        data=IssueResponse(
+            patch=patch,
+            passed_reproducing_test=passed_reproducing_test,
+            passed_build=passed_build,
+            passed_existing_test=passed_existing_test,
+            issue_response=issue_response,
+            remote_branch_name=remote_branch_name,
+        )
+    )