Sujatha commited on
Commit
a5b3bce
·
verified ·
1 Parent(s): 6764986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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.memory import ConversationBufferMemory
 
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 = ConversationBufferMemory(return_messages=True)
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.