From 606617ac728cb403efec17cb360493844ddaf4d8 Mon Sep 17 00:00:00 2001 From: xvnpw <17719543+xvnpw@users.noreply.github.com> Date: Sat, 7 Dec 2024 07:48:02 +0100 Subject: [PATCH] rename --- .github/workflows/ci.yaml | 6 ++-- .github/workflows/publish.yaml | 6 ++-- README.md | 34 +++++++++---------- .../__init__.py | 0 .../agent_builder.py | 14 ++++---- .../agents.py | 10 +++--- .../app.py | 12 +++---- .../base_agent.py | 6 ++-- .../config.py | 2 +- .../constants.py | 0 .../documents.py | 0 .../dry_run.py | 8 ++--- .../graphs.py | 2 +- .../llms.py | 4 +-- .../loaders.py | 0 .../markdowns.py | 0 .../utils.py | 0 entrypoint.sh | 2 +- examples/README.md | 8 ++--- package-lock.json | 2 +- pyproject.toml | 2 +- tests/test_agents.py | 12 +++---- tests/test_documents.py | 2 +- tests/test_graphs.py | 4 +-- tests/test_llms.py | 4 +-- tests/test_loaders.py | 2 +- tests/test_markdowns.py | 4 +-- 27 files changed, 73 insertions(+), 73 deletions(-) rename {ai_create_project_sec_design => ai_security_analyzer}/__init__.py (100%) rename {ai_create_project_sec_design => ai_security_analyzer}/agent_builder.py (78%) rename {ai_create_project_sec_design => ai_security_analyzer}/agents.py (98%) rename {ai_create_project_sec_design => ai_security_analyzer}/app.py (94%) rename {ai_create_project_sec_design => ai_security_analyzer}/base_agent.py (82%) rename {ai_create_project_sec_design => ai_security_analyzer}/config.py (97%) rename {ai_create_project_sec_design => ai_security_analyzer}/constants.py (100%) rename {ai_create_project_sec_design => ai_security_analyzer}/documents.py (100%) rename {ai_create_project_sec_design => ai_security_analyzer}/dry_run.py (90%) rename {ai_create_project_sec_design => ai_security_analyzer}/graphs.py (98%) rename {ai_create_project_sec_design => ai_security_analyzer}/llms.py (97%) rename {ai_create_project_sec_design => ai_security_analyzer}/loaders.py (100%) rename {ai_create_project_sec_design => ai_security_analyzer}/markdowns.py (100%) rename {ai_create_project_sec_design => ai_security_analyzer}/utils.py (100%) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 3b9b9bc..ba2da9e 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -32,7 +32,7 @@ jobs: run: poetry run ruff check --output-format=github . - name: Run MyPy - run: poetry run mypy ai_create_project_sec_design + run: poetry run mypy ai_security_analyzer - name: Test with pytest run: poetry run pytest -m "not integration" @@ -47,5 +47,5 @@ jobs: - name: Test docker run: | - docker build . -t test-ai-create-project-sec-design - docker run test-ai-create-project-sec-design -h + docker build . -t test-ai-security-analyzer + docker run test-ai-security-analyzer -h diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml index 4d95c9d..a6ab77d 100644 --- a/.github/workflows/publish.yaml +++ b/.github/workflows/publish.yaml @@ -42,7 +42,7 @@ jobs: run: poetry run ruff check --output-format=github . - name: Run MyPy - run: poetry run mypy ai_create_project_sec_design + run: poetry run mypy ai_security_analyzer - name: Test with pytest run: poetry run pytest -m "not integration" @@ -57,8 +57,8 @@ jobs: - name: Test docker run: | - docker build . -t test-ai-create-project-sec-design - docker run test-ai-create-project-sec-design -h + docker build . -t test-ai-security-analyzer + docker run test-ai-security-analyzer -h # Uses the `docker/login-action` action to log in to the Container registry registry using the account and password that will publish the packages. Once published, the packages are scoped to the account defined here. - name: Log in to the Container registry diff --git a/README.md b/README.md index 4021ed9..015aee4 100644 --- a/README.md +++ b/README.md @@ -1,17 +1,17 @@
- # AI Create Project Security Design + # AI Security Analyzer - + logo - [![CI](https://github.com/xvnpw/ai-create-project-sec-design/actions/workflows/ci.yaml/badge.svg)](https://github.com/xvnpw/ai-create-project-sec-design/actions/workflows/ci.yaml) - [![GitHub release](https://img.shields.io/github/release/xvnpw/ai-create-project-sec-design.svg)](https://github.com/xvnpw/ai-create-project-sec-design/releases) + [![CI](https://github.com/xvnpw/ai-security-analyzer/actions/workflows/ci.yaml/badge.svg)](https://github.com/xvnpw/ai-security-analyzer/actions/workflows/ci.yaml) + [![GitHub release](https://img.shields.io/github/release/xvnpw/ai-security-analyzer.svg)](https://github.com/xvnpw/ai-security-analyzer/releases) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT) - 🤖 **AI Create Project Security Design** is a powerful tool that leverages AI to automatically generate comprehensive security design documentation for your projects. + 🤖 **AI Security Analyzer** is a powerful tool that leverages AI to automatically generate comprehensive security design documentation for your projects. **🎥 Demo:** @@ -21,7 +21,7 @@ ## Overview -**AI Create Project Security Design** is a Python-based tool that analyzes your project's codebase and automatically generates detailed security design documentation. It supports multiple project types and utilizes advanced language models (LLMs) to create insightful security design documents tailored to your project's specific needs. +**AI Security Analyzer** is a Python-based tool that analyzes your project's codebase and automatically generates detailed security design documentation. It supports multiple project types and utilizes advanced language models (LLMs) to create insightful security design documents tailored to your project's specific needs. ## Features @@ -47,10 +47,10 @@ Clone the repository and install dependencies using the provided script: ```bash -git clone git@github.com:xvnpw/ai-create-project-sec-design.git -cd ai-create-project-sec-design +git clone git@github.com:xvnpw/ai-security-analyzer.git +cd ai-security-analyzer ./build.sh # Installs Python and Node.js dependencies -poetry run python ai_create_project_sec_design/app.py --help +poetry run python ai_security_analyzer/app.py --help ``` ### Using Docker @@ -62,7 +62,7 @@ You can run the application using Docker without installing Python or Node.js lo ```powershell docker run -v C:\path\to\your\project:/target ` -e OPENAI_API_KEY=$Env:OPENAI_API_KEY ` - ghcr.io/xvnpw/ai-create-project-sec-design:v0.0.1 ` + ghcr.io/xvnpw/ai-security-analyzer:v0.0.1 ` -v -t /target -o /target/security_design.md ``` @@ -71,7 +71,7 @@ docker run -v C:\path\to\your\project:/target ` ```bash docker run -v ~/path/to/your/project:/target \ -e OPENAI_API_KEY=$OPENAI_API_KEY \ - ghcr.io/xvnpw/ai-create-project-sec-design:v0.0.1 \ + ghcr.io/xvnpw/ai-security-analyzer:v0.0.1 \ -v -t /target -o /target/security_design.md ``` @@ -89,7 +89,7 @@ This application may consume a significant number of tokens due to its workflow: 1. **Always Start with Dry Run** ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ --dry-run ``` @@ -204,7 +204,7 @@ Set one of the following environment variables based on your chosen LLM provider Generate a security design document for a Python project: ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ -o security_design.md ``` @@ -214,7 +214,7 @@ poetry run python ai_create_project_sec_design/app.py \ Exclude the `tests` directory and `LICENSE` file: ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ -o security_design.md \ --exclude "LICENSE,**/tests/**" @@ -225,7 +225,7 @@ poetry run python ai_create_project_sec_design/app.py \ Use Anthropic's Claude model: ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ -o security_design.md \ --agent-provider anthropic \ @@ -239,7 +239,7 @@ poetry run python ai_create_project_sec_design/app.py \ See which files would be analyzed without making API calls: ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ --dry-run ``` @@ -273,7 +273,7 @@ This warning indicates that some document chunks exceed the LLM's context window **Example:** ```bash -poetry run python ai_create_project_sec_design/app.py \ +poetry run python ai_security_analyzer/app.py \ -t /path/to/your/project \ --files-chunk-size 50000 \ --files-context-window 70000 diff --git a/ai_create_project_sec_design/__init__.py b/ai_security_analyzer/__init__.py similarity index 100% rename from ai_create_project_sec_design/__init__.py rename to ai_security_analyzer/__init__.py diff --git a/ai_create_project_sec_design/agent_builder.py b/ai_security_analyzer/agent_builder.py similarity index 78% rename from ai_create_project_sec_design/agent_builder.py rename to ai_security_analyzer/agent_builder.py index a0efa4e..f04dda2 100644 --- a/ai_create_project_sec_design/agent_builder.py +++ b/ai_security_analyzer/agent_builder.py @@ -4,13 +4,13 @@ import tiktoken from langchain_text_splitters import CharacterTextSplitter -from ai_create_project_sec_design.agents import CreateProjectSecurityDesignAgent -from ai_create_project_sec_design.base_agent import BaseAgent -from ai_create_project_sec_design.config import AppConfig -from ai_create_project_sec_design.documents import DocumentFilter, DocumentProcessor -from ai_create_project_sec_design.dry_run import DryRunAgent -from ai_create_project_sec_design.llms import LLMProvider -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.agents import CreateProjectSecurityDesignAgent +from ai_security_analyzer.base_agent import BaseAgent +from ai_security_analyzer.config import AppConfig +from ai_security_analyzer.documents import DocumentFilter, DocumentProcessor +from ai_security_analyzer.dry_run import DryRunAgent +from ai_security_analyzer.llms import LLMProvider +from ai_security_analyzer.markdowns import MarkdownMermaidValidator logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/agents.py b/ai_security_analyzer/agents.py similarity index 98% rename from ai_create_project_sec_design/agents.py rename to ai_security_analyzer/agents.py index da3476d..5a827c3 100644 --- a/ai_create_project_sec_design/agents.py +++ b/ai_security_analyzer/agents.py @@ -12,11 +12,11 @@ from langgraph.graph.state import CompiledStateGraph from tiktoken import Encoding -from ai_create_project_sec_design.base_agent import BaseAgent -from ai_create_project_sec_design.documents import DocumentFilter, DocumentProcessor -from ai_create_project_sec_design.llms import LLMProvider -from ai_create_project_sec_design.loaders import RepoDirectoryLoader -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.base_agent import BaseAgent +from ai_security_analyzer.documents import DocumentFilter, DocumentProcessor +from ai_security_analyzer.llms import LLMProvider +from ai_security_analyzer.loaders import RepoDirectoryLoader +from ai_security_analyzer.markdowns import MarkdownMermaidValidator logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/app.py b/ai_security_analyzer/app.py similarity index 94% rename from ai_create_project_sec_design/app.py rename to ai_security_analyzer/app.py index bad8c57..b0d6c41 100644 --- a/ai_create_project_sec_design/app.py +++ b/ai_security_analyzer/app.py @@ -2,10 +2,10 @@ import logging import sys -from ai_create_project_sec_design.agent_builder import AgentBuilder -from ai_create_project_sec_design.config import AppConfig -from ai_create_project_sec_design.graphs import GraphExecutorFactory -from ai_create_project_sec_design.llms import LLMProvider +from ai_security_analyzer.agent_builder import AgentBuilder +from ai_security_analyzer.config import AppConfig +from ai_security_analyzer.graphs import GraphExecutorFactory +from ai_security_analyzer.llms import LLMProvider logger = logging.getLogger(__name__) @@ -195,11 +195,11 @@ def main() -> None: config = parse_arguments() setup_logging(config.verbose, config.debug) - logger.info("Starting AI Create Project Security Design") + logger.info("Starting AI Security Analyzer") app(config) - logger.info("AI Create Project Security Design completed successfully") + logger.info("AI Security Analyzer completed successfully") except Exception as e: logger.error(f"Application error: {e}") diff --git a/ai_create_project_sec_design/base_agent.py b/ai_security_analyzer/base_agent.py similarity index 82% rename from ai_create_project_sec_design/base_agent.py rename to ai_security_analyzer/base_agent.py index 7236e26..60ce5ea 100644 --- a/ai_create_project_sec_design/base_agent.py +++ b/ai_security_analyzer/base_agent.py @@ -5,9 +5,9 @@ from langchain_text_splitters import CharacterTextSplitter from langgraph.graph.state import CompiledStateGraph -from ai_create_project_sec_design.documents import DocumentFilter, DocumentProcessor -from ai_create_project_sec_design.llms import LLMProvider -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.documents import DocumentFilter, DocumentProcessor +from ai_security_analyzer.llms import LLMProvider +from ai_security_analyzer.markdowns import MarkdownMermaidValidator logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/config.py b/ai_security_analyzer/config.py similarity index 97% rename from ai_create_project_sec_design/config.py rename to ai_security_analyzer/config.py index e48c086..89bd52d 100644 --- a/ai_create_project_sec_design/config.py +++ b/ai_security_analyzer/config.py @@ -4,7 +4,7 @@ from pydantic import BaseModel, ConfigDict, Field, field_validator from typing_extensions import Literal -from ai_create_project_sec_design.utils import find_node_binary +from ai_security_analyzer.utils import find_node_binary class AppConfig(BaseModel): diff --git a/ai_create_project_sec_design/constants.py b/ai_security_analyzer/constants.py similarity index 100% rename from ai_create_project_sec_design/constants.py rename to ai_security_analyzer/constants.py diff --git a/ai_create_project_sec_design/documents.py b/ai_security_analyzer/documents.py similarity index 100% rename from ai_create_project_sec_design/documents.py rename to ai_security_analyzer/documents.py diff --git a/ai_create_project_sec_design/dry_run.py b/ai_security_analyzer/dry_run.py similarity index 90% rename from ai_create_project_sec_design/dry_run.py rename to ai_security_analyzer/dry_run.py index 7ecedf7..32e0224 100644 --- a/ai_create_project_sec_design/dry_run.py +++ b/ai_security_analyzer/dry_run.py @@ -5,13 +5,13 @@ from langgraph.graph.state import CompiledStateGraph from tiktoken import Encoding -from ai_create_project_sec_design.agents import ( +from ai_security_analyzer.agents import ( CreateProjectSecurityDesignAgent, AgentState, ) -from ai_create_project_sec_design.documents import DocumentFilter, DocumentProcessor -from ai_create_project_sec_design.llms import LLMProvider -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.documents import DocumentFilter, DocumentProcessor +from ai_security_analyzer.llms import LLMProvider +from ai_security_analyzer.markdowns import MarkdownMermaidValidator logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/graphs.py b/ai_security_analyzer/graphs.py similarity index 98% rename from ai_create_project_sec_design/graphs.py rename to ai_security_analyzer/graphs.py index 6492d66..0a96539 100644 --- a/ai_create_project_sec_design/graphs.py +++ b/ai_security_analyzer/graphs.py @@ -3,7 +3,7 @@ from typing import Any, Final, List from langgraph.graph.state import CompiledStateGraph -from ai_create_project_sec_design.config import AppConfig +from ai_security_analyzer.config import AppConfig from langchain_core.documents import Document logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/llms.py b/ai_security_analyzer/llms.py similarity index 97% rename from ai_create_project_sec_design/llms.py rename to ai_security_analyzer/llms.py index 30f90a8..e698257 100644 --- a/ai_create_project_sec_design/llms.py +++ b/ai_security_analyzer/llms.py @@ -8,8 +8,8 @@ from langchain_core.language_models.chat_models import BaseChatModel from langchain_openai import ChatOpenAI -from ai_create_project_sec_design import constants -from ai_create_project_sec_design.config import AppConfig +from ai_security_analyzer import constants +from ai_security_analyzer.config import AppConfig logger = logging.getLogger(__name__) diff --git a/ai_create_project_sec_design/loaders.py b/ai_security_analyzer/loaders.py similarity index 100% rename from ai_create_project_sec_design/loaders.py rename to ai_security_analyzer/loaders.py diff --git a/ai_create_project_sec_design/markdowns.py b/ai_security_analyzer/markdowns.py similarity index 100% rename from ai_create_project_sec_design/markdowns.py rename to ai_security_analyzer/markdowns.py diff --git a/ai_create_project_sec_design/utils.py b/ai_security_analyzer/utils.py similarity index 100% rename from ai_create_project_sec_design/utils.py rename to ai_security_analyzer/utils.py diff --git a/entrypoint.sh b/entrypoint.sh index efeb377..1279302 100644 --- a/entrypoint.sh +++ b/entrypoint.sh @@ -8,4 +8,4 @@ if [ -n "$GITHUB_WORKSPACE" ]; then echo "Detected GitHub runner" fi -exec /usr/local/bin/python /app/ai_create_project_sec_design/app.py "$@" +exec /usr/local/bin/python /app/ai_security_analyzer/app.py "$@" diff --git a/examples/README.md b/examples/README.md index 44b33bc..00f4e6c 100644 --- a/examples/README.md +++ b/examples/README.md @@ -2,7 +2,7 @@ | Name | Type | File | | --- | ---| --- | -| [caddy](https://github.com/caddyserver/caddy) - Fast and extensible multi-platform HTTP/1-2-3 web server with automatic HTTPS
command...`python ai_create_project_sec_design/app.py -t caddy/ -v --project-type go -o examples/CADDY-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| go | [o1-preview](./CADDY-o1-preview.md), [gpt-4o](./CADDY-gpt-4o.md), [threat model - o1-preview](./TM-CADDY-o1-preview.md) -| [screenshot-to-code](https://github.com/abi/screenshot-to-code) - Drop in a screenshot and convert it to clean code (HTML/Tailwind/React/Vue)
command...`python ai_create_project_sec_design/app.py -t screenshot-to-code/ -v -o examples/SCREENSHOT_TO_CODE-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [o1-preview](./SCREENSHOT-TO-CODE-o1-preview.md), [gpt-4o](./SCREENSHOT-TO-CODE-gpt-4o.md), [threat model - o1-preview](./TM-SCREENSHOT-TO-CODE-o1-preview.md) | -| [requests](https://github.com/psf/requests) - A simple, yet elegant, HTTP library
command...`python ai_create_project_sec_design/app.py -t requests/ -v --exclude "**/ISSUE_TEMPLATE*,**/CODE_OF_CONDUCT.md,**/CONTRIBUTING.md,**/FUNDING.yml" --include "**/*.cfg,**/*.rst" -o examples/REQUESTS-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [gpt-4o](./REQUESTS-gpt-4o.md), [o1-preview](./REQUESTS-o1-preview.md), [threat model - o1-preview](./TM-REQUESTS-o1-preview.md) | -| [flask](https://github.com/pallets/flask) - The Python micro framework for building web applications
command...`python ai_create_project_sec_design/app.py -t ../flask/ -v --exclude "**/pull_request_template.md,**/ISSUE_TEMPLATE*,**/CODE_OF_CONDUCT.md" --include "**/requirements/*.txt,**/*.rst" -o examples/FLASK-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [gpt-4o](./FLASK-gpt-4o.md), [o1-preview](./FLASK-o1-preview.md), [threat model - o1-preview](./TM-FLASK-o1-preview.md) | +| [caddy](https://github.com/caddyserver/caddy) - Fast and extensible multi-platform HTTP/1-2-3 web server with automatic HTTPS
command...`python ai_security_analyzer/app.py -t caddy/ -v --project-type go -o examples/CADDY-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| go | [o1-preview](./CADDY-o1-preview.md), [gpt-4o](./CADDY-gpt-4o.md), [threat model - o1-preview](./TM-CADDY-o1-preview.md) +| [screenshot-to-code](https://github.com/abi/screenshot-to-code) - Drop in a screenshot and convert it to clean code (HTML/Tailwind/React/Vue)
command...`python ai_security_analyzer/app.py -t screenshot-to-code/ -v -o examples/SCREENSHOT_TO_CODE-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [o1-preview](./SCREENSHOT-TO-CODE-o1-preview.md), [gpt-4o](./SCREENSHOT-TO-CODE-gpt-4o.md), [threat model - o1-preview](./TM-SCREENSHOT-TO-CODE-o1-preview.md) | +| [requests](https://github.com/psf/requests) - A simple, yet elegant, HTTP library
command...`python ai_security_analyzer/app.py -t requests/ -v --exclude "**/ISSUE_TEMPLATE*,**/CODE_OF_CONDUCT.md,**/CONTRIBUTING.md,**/FUNDING.yml" --include "**/*.cfg,**/*.rst" -o examples/REQUESTS-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [gpt-4o](./REQUESTS-gpt-4o.md), [o1-preview](./REQUESTS-o1-preview.md), [threat model - o1-preview](./TM-REQUESTS-o1-preview.md) | +| [flask](https://github.com/pallets/flask) - The Python micro framework for building web applications
command...`python ai_security_analyzer/app.py -t ../flask/ -v --exclude "**/pull_request_template.md,**/ISSUE_TEMPLATE*,**/CODE_OF_CONDUCT.md" --include "**/requirements/*.txt,**/*.rst" -o examples/FLASK-o1-preview.md --agent-model o1-preview --agent-temperature 1`
| python | [gpt-4o](./FLASK-gpt-4o.md), [o1-preview](./FLASK-o1-preview.md), [threat model - o1-preview](./TM-FLASK-o1-preview.md) | diff --git a/package-lock.json b/package-lock.json index bbe6fe1..0af6ce7 100644 --- a/package-lock.json +++ b/package-lock.json @@ -1,5 +1,5 @@ { - "name": "ai-create-project-sec-design", + "name": "ai-security-analyzer", "lockfileVersion": 3, "requires": true, "packages": { diff --git a/pyproject.toml b/pyproject.toml index 8191b9f..18027bb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [tool.poetry] -name = "ai-create-project-sec-design" +name = "ai-security-analyzer" version = "0.0.1" description = "" authors = ["xvnpw <17719543+xvnpw@users.noreply.github.com>"] diff --git a/tests/test_agents.py b/tests/test_agents.py index ba24b82..052fd56 100644 --- a/tests/test_agents.py +++ b/tests/test_agents.py @@ -4,15 +4,15 @@ import pytest -from ai_create_project_sec_design.agent_builder import AgentBuilder -from ai_create_project_sec_design.agents import ( +from ai_security_analyzer.agent_builder import AgentBuilder +from ai_security_analyzer.agents import ( CreateProjectSecurityDesignAgent, GraphNodeType, ) -from ai_create_project_sec_design.base_agent import BaseAgent -from ai_create_project_sec_design.documents import DocumentFilter, DocumentProcessor -from ai_create_project_sec_design.llms import LLM, LLMProvider, ModelConfig -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.base_agent import BaseAgent +from ai_security_analyzer.documents import DocumentFilter, DocumentProcessor +from ai_security_analyzer.llms import LLM, LLMProvider, ModelConfig +from ai_security_analyzer.markdowns import MarkdownMermaidValidator from langchain_core.documents import Document diff --git a/tests/test_documents.py b/tests/test_documents.py index 293e12b..6db9f04 100644 --- a/tests/test_documents.py +++ b/tests/test_documents.py @@ -1,6 +1,6 @@ from langchain_core.documents import Document -from ai_create_project_sec_design.documents import DocumentProcessor, DocumentFilter +from ai_security_analyzer.documents import DocumentProcessor, DocumentFilter # Mock Tokenizer for testing diff --git a/tests/test_graphs.py b/tests/test_graphs.py index 8dafb77..b888edf 100644 --- a/tests/test_graphs.py +++ b/tests/test_graphs.py @@ -1,6 +1,6 @@ from unittest.mock import Mock -from ai_create_project_sec_design.graphs import RunGraphExecutor, DryRunGraphExecutor, GraphExecutorFactory -from ai_create_project_sec_design.config import AppConfig +from ai_security_analyzer.graphs import RunGraphExecutor, DryRunGraphExecutor, GraphExecutorFactory +from ai_security_analyzer.config import AppConfig from langgraph.graph.state import CompiledStateGraph from langchain_core.documents import Document diff --git a/tests/test_llms.py b/tests/test_llms.py index 67a7452..e5d8f39 100644 --- a/tests/test_llms.py +++ b/tests/test_llms.py @@ -5,8 +5,8 @@ import pytest from langchain_core.language_models.chat_models import BaseChatModel -from ai_create_project_sec_design import constants -from ai_create_project_sec_design.llms import LLMProvider +from ai_security_analyzer import constants +from ai_security_analyzer.llms import LLMProvider @dataclass diff --git a/tests/test_loaders.py b/tests/test_loaders.py index f4984f8..bb7cba3 100644 --- a/tests/test_loaders.py +++ b/tests/test_loaders.py @@ -1,6 +1,6 @@ from pathlib import Path -from ai_create_project_sec_design.loaders import RepoDirectoryLoader +from ai_security_analyzer.loaders import RepoDirectoryLoader def _in_path(substring, paths): diff --git a/tests/test_markdowns.py b/tests/test_markdowns.py index 69f2d9a..1bae2ce 100644 --- a/tests/test_markdowns.py +++ b/tests/test_markdowns.py @@ -1,5 +1,5 @@ -from ai_create_project_sec_design.markdowns import MarkdownMermaidValidator -from ai_create_project_sec_design.utils import find_node_binary +from ai_security_analyzer.markdowns import MarkdownMermaidValidator +from ai_security_analyzer.utils import find_node_binary NODE_PATH = find_node_binary() if not NODE_PATH: