fix imports

pyproject.toml

@@ -29,4 +29,4 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 package-mode = true
-packages = [{ include = "src/upgrade_advisor" }]
+packages = [{ include = "upgrade_advisor", from = "src" }]

requirements.txt

@@ -1,308 +0,0 @@
-# This file was autogenerated by uv via the following command:
-#    uv pip compile pyproject.toml -o requirements.txt
-aiofiles==24.1.0
-    # via gradio
-annotated-doc==0.0.4
-    # via fastapi
-annotated-types==0.7.0
-    # via pydantic
-anyio==4.12.0
-    # via
-    #   upgrade-advisor (pyproject.toml)
-    #   gradio
-    #   httpx
-    #   mcp
-    #   sse-starlette
-    #   starlette
-attrs==25.4.0
-    # via
-    #   jsonschema
-    #   referencing
-authlib==1.6.5
-    # via gradio
-backports-asyncio-runner==1.2.0
-    # via pytest-asyncio
-beautifulsoup4==4.14.2
-    # via markdownify
-brotli==1.2.0
-    # via
-    #   gradio
-    #   httpx
-certifi==2025.11.12
-    # via
-    #   httpcore
-    #   httpx
-    #   requests
-cffi==2.0.0
-    # via cryptography
-charset-normalizer==3.4.4
-    # via requests
-click==8.3.1
-    # via
-    #   ddgs
-    #   typer
-    #   uvicorn
-cryptography==46.0.3
-    # via
-    #   authlib
-    #   pyjwt
-ddgs==9.9.2
-    # via upgrade-advisor (pyproject.toml)
-exceptiongroup==1.3.1
-    # via
-    #   anyio
-    #   pytest
-fake-useragent==2.2.0
-    # via ddgs
-fastapi==0.122.1
-    # via gradio
-ffmpy==1.0.0
-    # via gradio
-filelock==3.20.0
-    # via
-    #   huggingface-hub
-    #   transformers
-fsspec==2025.10.0
-    # via
-    #   gradio-client
-    #   huggingface-hub
-gradio==6.0.1
-    # via upgrade-advisor (pyproject.toml)
-gradio-client==2.0.0
-    # via gradio
-groovy==0.1.2
-    # via gradio
-h11==0.16.0
-    # via
-    #   httpcore
-    #   uvicorn
-h2==4.3.0
-    # via httpx
-hf-xet==1.2.0
-    # via huggingface-hub
-hpack==4.1.0
-    # via h2
-httpcore==1.0.9
-    # via httpx
-httpx==0.28.1
-    # via
-    #   ddgs
-    #   gradio
-    #   gradio-client
-    #   mcp
-    #   safehttpx
-httpx-sse==0.4.3
-    # via mcp
-huggingface-hub==0.36.0
-    # via
-    #   gradio
-    #   gradio-client
-    #   smolagents
-    #   tokenizers
-    #   transformers
-hyperframe==6.1.0
-    # via h2
-idna==3.11
-    # via
-    #   anyio
-    #   httpx
-    #   requests
-iniconfig==2.3.0
-    # via pytest
-itsdangerous==2.2.0
-    # via gradio
-jinja2==3.1.6
-    # via
-    #   gradio
-    #   smolagents
-jsonref==1.1.0
-    # via mcpadapt
-jsonschema==4.25.1
-    # via mcp
-jsonschema-specifications==2025.9.1
-    # via jsonschema
-lxml==6.0.2
-    # via ddgs
-markdown-it-py==4.0.0
-    # via rich
-markdownify==1.2.2
-    # via upgrade-advisor (pyproject.toml)
-markupsafe==3.0.3
-    # via
-    #   gradio
-    #   jinja2
-mcp==1.22.0
-    # via
-    #   upgrade-advisor (pyproject.toml)
-    #   mcpadapt
-    #   smolagents
-mcpadapt==0.1.19
-    # via smolagents
-mdurl==0.1.2
-    # via markdown-it-py
-numpy==2.2.6
-    # via
-    #   gradio
-    #   pandas
-    #   transformers
-orjson==3.11.4
-    # via gradio
-packaging==25.0
-    # via
-    #   gradio
-    #   gradio-client
-    #   huggingface-hub
-    #   pytest
-    #   transformers
-pandas==2.3.3
-    # via gradio
-pillow==12.0.0
-    # via
-    #   gradio
-    #   smolagents
-pluggy==1.6.0
-    # via pytest
-primp==0.15.0
-    # via ddgs
-pycparser==2.23
-    # via cffi
-pydantic==2.12.4
-    # via
-    #   fastapi
-    #   gradio
-    #   mcp
-    #   mcpadapt
-    #   pydantic-settings
-pydantic-core==2.41.5
-    # via pydantic
-pydantic-settings==2.12.0
-    # via mcp
-pydub==0.25.1
-    # via gradio
-pygments==2.19.2
-    # via
-    #   pytest
-    #   rich
-pyjwt==2.10.1
-    # via mcp
-pytest==9.0.1
-    # via
-    #   upgrade-advisor (pyproject.toml)
-    #   pytest-asyncio
-pytest-asyncio==1.3.0
-    # via upgrade-advisor (pyproject.toml)
-python-dateutil==2.9.0.post0
-    # via pandas
-python-dotenv==1.2.1
-    # via
-    #   upgrade-advisor (pyproject.toml)
-    #   mcpadapt
-    #   pydantic-settings
-    #   smolagents
-python-multipart==0.0.20
-    # via
-    #   gradio
-    #   mcp
-pytz==2025.2
-    # via pandas
-pyyaml==6.0.3
-    # via
-    #   gradio
-    #   huggingface-hub
-    #   transformers
-referencing==0.37.0
-    # via
-    #   jsonschema
-    #   jsonschema-specifications
-regex==2025.11.3
-    # via transformers
-requests==2.32.5
-    # via
-    #   huggingface-hub
-    #   smolagents
-    #   transformers
-rich==14.2.0
-    # via
-    #   smolagents
-    #   typer
-rpds-py==0.29.0
-    # via
-    #   jsonschema
-    #   referencing
-safehttpx==0.1.7
-    # via gradio
-safetensors==0.7.0
-    # via transformers
-semantic-version==2.10.0
-    # via gradio
-shellingham==1.5.4
-    # via typer
-six==1.17.0
-    # via
-    #   markdownify
-    #   python-dateutil
-smolagents==1.23.0
-    # via upgrade-advisor (pyproject.toml)
-socksio==1.0.0
-    # via httpx
-soupsieve==2.8
-    # via beautifulsoup4
-sse-starlette==3.0.3
-    # via mcp
-starlette==0.50.0
-    # via
-    #   fastapi
-    #   gradio
-    #   mcp
-tokenizers==0.22.1
-    # via transformers
-tomli==2.3.0
-    # via pytest
-tomlkit==0.13.3
-    # via gradio
-tqdm==4.67.1
-    # via
-    #   huggingface-hub
-    #   transformers
-transformers==4.57.3
-    # via upgrade-advisor (pyproject.toml)
-typer==0.20.0
-    # via
-    #   upgrade-advisor (pyproject.toml)
-    #   gradio
-typing-extensions==4.15.0
-    # via
-    #   anyio
-    #   beautifulsoup4
-    #   cryptography
-    #   exceptiongroup
-    #   fastapi
-    #   gradio
-    #   gradio-client
-    #   huggingface-hub
-    #   mcp
-    #   pydantic
-    #   pydantic-core
-    #   pytest-asyncio
-    #   referencing
-    #   starlette
-    #   typer
-    #   typing-inspection
-    #   uvicorn
-typing-inspection==0.4.2
-    # via
-    #   mcp
-    #   pydantic
-    #   pydantic-settings
-tzdata==2025.2
-    # via pandas
-urllib3==2.5.0
-    # via requests
-uv==0.9.13
-    # via upgrade-advisor (pyproject.toml)
-uvicorn==0.38.0
-    # via
-    #   gradio
-    #   mcp
-websockets==15.0.1
-    # via mcp

src/upgrade_advisor/agents/tools/parse_response.py

@@ -1,6 +1,6 @@
 import logging
 
-from src.upgrade_advisor.schema import (
+from upgrade_advisor.schema import (
     PackageInfoSchema,
     PackageReleaseSchema,
     PackageSearchResponseSchema,

src/upgrade_advisor/agents/tools/pypi_api.py

@@ -4,7 +4,7 @@ from typing import Optional
 import requests
 from requests import HTTPError
 
-from src.upgrade_advisor.schema import (
+from upgrade_advisor.schema import (
     ErrorResponseSchema,
     GithubRepoSchema,
     PackageGitHubandReleasesSchema,

src/upgrade_advisor/agents/tools/tools.py

@@ -4,8 +4,8 @@ from pathlib import Path
 
 from smolagents.tools import Tool
 
-from src.upgrade_advisor.const import ALLOWED_OS, UPLOADS_DIR
-from src.upgrade_advisor.schema import (
+from upgrade_advisor.const import ALLOWED_OS, UPLOADS_DIR
+from upgrade_advisor.schema import (
     GithubRepoSchema,
     PackageGitHubandReleasesSchema,
     PackageSearchResponseSchema,

src/upgrade_advisor/agents/tools/uv_resolver.py

@@ -6,9 +6,9 @@ import tempfile
 from contextlib import contextmanager
 from typing import Literal
 
-from src.upgrade_advisor.agents.tools.parse_response import parse_resolved_deps
-from src.upgrade_advisor.const import ALLOWED_OS, UV_VERSION
-from src.upgrade_advisor.schema import (
+from upgrade_advisor.agents.tools.parse_response import parse_resolved_deps
+from upgrade_advisor.const import ALLOWED_OS, UV_VERSION
+from upgrade_advisor.schema import (
     ResolvedDep,
     ResolveResult,
     UVResolutionResultSchema,

src/upgrade_advisor/app.py

@@ -1,6 +1,7 @@
 import logging
 import os
 import shutil
+import sys
 from pathlib import Path
 
 import gradio as gr
@@ -8,7 +9,7 @@ from mcp import StdioServerParameters
 from smolagents import InferenceClientModel
 from smolagents.mcp_client import MCPClient
 
-from config import (
+from config import (  # noqa: E402
     AGENT_MODEL,
     CHAT_HISTORY_TURNS_CUTOFF,
     CHAT_HISTORY_WORD_CUTOFF,
@@ -17,17 +18,17 @@ from config import (
     GITHUB_TOOLSETS,
 )
 
-from .agents.package import PackageDiscoveryAgent
-from .chat.chat import (
+from upgrade_advisor.agents.package import PackageDiscoveryAgent  # noqa: E402
+from upgrade_advisor.chat.chat import (  # noqa: E402
     qn_rewriter,
     run_document_qa,
     summarize_chat_history,
 )
-from .misc import (
+from upgrade_advisor.misc import (  # noqa: E402
     _monkeypatch_gradio_save_history,
     get_example_questions,
 )
-from .theme import Christmas
+from upgrade_advisor.theme import Christmas  # noqa: E402
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)