LiamKhoaLe commited on
Commit
7cf24a1
·
1 Parent(s): e2385d4

Upd imports

Browse files
Files changed (4) hide show
  1. api/app.py +3 -3
  2. api/chatbot.py +2 -2
  3. api/routes.py +2 -2
  4. models/llama.py +1 -1
api/app.py CHANGED
@@ -2,9 +2,9 @@
2
  import uvicorn
3
  from fastapi import FastAPI
4
  from fastapi.middleware.cors import CORSMiddleware
5
- from api.config import setup_logging, check_system_resources, optimize_memory, CORS_ORIGINS, validate_environment
6
- from api.database import db_manager
7
- from api.routes import router
8
 
9
  # ✅ Validate environment
10
  validate_environment()
 
2
  import uvicorn
3
  from fastapi import FastAPI
4
  from fastapi.middleware.cors import CORSMiddleware
5
+ from config import setup_logging, check_system_resources, optimize_memory, CORS_ORIGINS, validate_environment
6
+ from database import db_manager
7
+ from routes import router
8
 
9
  # ✅ Validate environment
10
  validate_environment()
api/chatbot.py CHANGED
@@ -3,8 +3,8 @@ import re
3
  import logging
4
  from typing import Dict
5
  from google import genai
6
- from api.config import gemini_flash_api_key
7
- from api.retrieval import retrieval_engine
8
  from memory import MemoryManager
9
  from utils import translate_query, process_medical_image
10
  from search import search_web
 
3
  import logging
4
  from typing import Dict
5
  from google import genai
6
+ from config import gemini_flash_api_key
7
+ from retrieval import retrieval_engine
8
  from memory import MemoryManager
9
  from utils import translate_query, process_medical_image
10
  from search import search_web
api/routes.py CHANGED
@@ -3,8 +3,8 @@ import time
3
  import logging
4
  from fastapi import APIRouter, Request
5
  from fastapi.responses import JSONResponse
6
- from api.chatbot import RAGMedicalChatbot
7
- from api.retrieval import retrieval_engine
8
  from utils import process_medical_image
9
 
10
  logger = logging.getLogger("medical-chatbot")
 
3
  import logging
4
  from fastapi import APIRouter, Request
5
  from fastapi.responses import JSONResponse
6
+ from chatbot import RAGMedicalChatbot
7
+ from retrieval import retrieval_engine
8
  from utils import process_medical_image
9
 
10
  logger = logging.getLogger("medical-chatbot")
models/llama.py CHANGED
@@ -42,7 +42,7 @@ Keywords:"""
42
  """Use Llama to summarize documents and return summary with URL mapping"""
43
  try:
44
  # Import summarizer here to avoid circular imports
45
- from models.summarizer import summarizer
46
 
47
  # Use the summarizer for document summarization
48
  combined_summary, url_mapping = summarizer.summarize_documents(documents, user_query)
 
42
  """Use Llama to summarize documents and return summary with URL mapping"""
43
  try:
44
  # Import summarizer here to avoid circular imports
45
+ from summarizer import summarizer
46
 
47
  # Use the summarizer for document summarization
48
  combined_summary, url_mapping = summarizer.summarize_documents(documents, user_query)