Rashid Ali commited on
Commit
10b1f68
·
1 Parent(s): a2dd353

chatstate fix

Browse files
Files changed (1) hide show
  1. src/chatbot_backend.py +2 -2
src/chatbot_backend.py CHANGED
@@ -27,8 +27,8 @@ class ChatState(TypedDict):
27
  doc_content: Optional[str] # User's uploaded document content
28
  search_enabled: bool # Whether to use external search
29
  model_choice: str # Model selected ("OpenAI (Paid)" or "LLaMA (Open Source)")
30
- doc_source: str | None # Document source ("Upload Your Documents" or "Use Sample Documents")
31
- system_prompt: str | None # ✅ New field for system prompt
32
 
33
  # -------------------
34
  # 3. Chat Node (Main LLM Node)
 
27
  doc_content: Optional[str] # User's uploaded document content
28
  search_enabled: bool # Whether to use external search
29
  model_choice: str # Model selected ("OpenAI (Paid)" or "LLaMA (Open Source)")
30
+ doc_source: Optional[str] # Document source ("Upload Your Documents" or "Use Sample Documents")
31
+ system_prompt: Optional[str] # ✅ New field for system prompt
32
 
33
  # -------------------
34
  # 3. Chat Node (Main LLM Node)