Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,8 @@ import json
|
|
| 8 |
from typing import Dict, List, Tuple, Optional
|
| 9 |
|
| 10 |
# LangChain imports for conversational AI
|
| 11 |
-
from langchain.
|
|
|
|
| 12 |
from langchain.schema import HumanMessage, AIMessage, SystemMessage
|
| 13 |
from langchain_openai import ChatOpenAI
|
| 14 |
|
|
@@ -16,8 +17,8 @@ from langchain_openai import ChatOpenAI
|
|
| 16 |
from index_calculator import IndexCalculator
|
| 17 |
from data_sources import RealTimeIndexCalculator, IndianStockAPI, GlobalStockAPI
|
| 18 |
|
| 19 |
-
# Initialize memory for conversation context
|
| 20 |
-
memory =
|
| 21 |
|
| 22 |
# System prompt for the AI assistant
|
| 23 |
SYSTEM_PROMPT = """You are an expert financial analyst specializing in index numbers and real-time market data.
|
|
|
|
| 8 |
from typing import Dict, List, Tuple, Optional
|
| 9 |
|
| 10 |
# LangChain imports for conversational AI
|
| 11 |
+
from langchain.chains import ConversationChain # Replaced ConversationBufferMemory with ConversationChain
|
| 12 |
+
from langchain.prompts import PromptTemplate
|
| 13 |
from langchain.schema import HumanMessage, AIMessage, SystemMessage
|
| 14 |
from langchain_openai import ChatOpenAI
|
| 15 |
|
|
|
|
| 17 |
from index_calculator import IndexCalculator
|
| 18 |
from data_sources import RealTimeIndexCalculator, IndianStockAPI, GlobalStockAPI
|
| 19 |
|
| 20 |
+
# Initialize memory for conversation context using ConversationChain (if ConversationBufferMemory is not found)
|
| 21 |
+
memory = ConversationChain(memory_type="buffer", return_messages=True) # Adjusted for compatibility
|
| 22 |
|
| 23 |
# System prompt for the AI assistant
|
| 24 |
SYSTEM_PROMPT = """You are an expert financial analyst specializing in index numbers and real-time market data.
|