Geevarghese George commited on
Commit
9c244cf
·
1 Parent(s): d679175
Files changed (2) hide show
  1. requirements.txt +1 -2
  2. src/upgrade_advisor/app.py +5 -5
requirements.txt CHANGED
@@ -9,5 +9,4 @@ transformers
9
  anyio
10
  pytest-asyncio
11
  pytest
12
- mcp
13
- .
 
9
  anyio
10
  pytest-asyncio
11
  pytest
12
+ mcp
 
src/upgrade_advisor/app.py CHANGED
@@ -9,7 +9,7 @@ from mcp import StdioServerParameters
9
  from smolagents import InferenceClientModel
10
  from smolagents.mcp_client import MCPClient
11
 
12
- from upgrade_advisor.config import ( # noqa: E402
13
  AGENT_MODEL,
14
  CHAT_HISTORY_TURNS_CUTOFF,
15
  CHAT_HISTORY_WORD_CUTOFF,
@@ -18,17 +18,17 @@ from upgrade_advisor.config import ( # noqa: E402
18
  GITHUB_TOOLSETS,
19
  )
20
 
21
- from upgrade_advisor.agents.package import PackageDiscoveryAgent # noqa: E402
22
- from upgrade_advisor.chat.chat import ( # noqa: E402
23
  qn_rewriter,
24
  run_document_qa,
25
  summarize_chat_history,
26
  )
27
- from upgrade_advisor.misc import ( # noqa: E402
28
  _monkeypatch_gradio_save_history,
29
  get_example_questions,
30
  )
31
- from upgrade_advisor.theme import Christmas # noqa: E402
32
 
33
  logger = logging.getLogger(__name__)
34
  logger.setLevel(logging.INFO)
 
9
  from smolagents import InferenceClientModel
10
  from smolagents.mcp_client import MCPClient
11
 
12
+ from .config import ( # noqa: E402
13
  AGENT_MODEL,
14
  CHAT_HISTORY_TURNS_CUTOFF,
15
  CHAT_HISTORY_WORD_CUTOFF,
 
18
  GITHUB_TOOLSETS,
19
  )
20
 
21
+ from .agents.package import PackageDiscoveryAgent # noqa: E402
22
+ from .chat.chat import ( # noqa: E402
23
  qn_rewriter,
24
  run_document_qa,
25
  summarize_chat_history,
26
  )
27
+ from .misc import ( # noqa: E402
28
  _monkeypatch_gradio_save_history,
29
  get_example_questions,
30
  )
31
+ from .theme import Christmas # noqa: E402
32
 
33
  logger = logging.getLogger(__name__)
34
  logger.setLevel(logging.INFO)