Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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
|
| 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
|