pvanand commited on
Commit
d017278
·
verified ·
1 Parent(s): fd86aa0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from pydantic import BaseModel
4
  from typing import List, Dict, Optional, Union, Annotated, Any, get_args
5
  from openai import AsyncOpenAI
6
  from observability import LLMObservabilityManager, log_execution, logger
7
- from aiclient import DatabaseManager, AIClient
8
  from limit_tokens import trim_messages_openai
9
  from prompts import FOLLOWUP_DIGIYATRA_PROMPT
10
  from utils import parse_followup_and_tools
 
4
  from typing import List, Dict, Optional, Union, Annotated, Any, get_args
5
  from openai import AsyncOpenAI
6
  from observability import LLMObservabilityManager, log_execution, logger
7
+ from aws_aiclient import DatabaseManager, AIClient
8
  from limit_tokens import trim_messages_openai
9
  from prompts import FOLLOWUP_DIGIYATRA_PROMPT
10
  from utils import parse_followup_and_tools