File size: 936 Bytes
1ef298a c1a68be 1ef298a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from typing import List
# from backend.app.vectorstore import get_vector_db
from langchain.agents import AgentExecutor, create_openai_functions_agent
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain.output_parsers.openai_functions import JsonOutputFunctionsParser
from langchain_openai import ChatOpenAI
class ProblemGenerator:
def generate_problems(self, query: str) -> List[str]:
"""
Generate problems based on the user's query.
"""
# For MVP, returning random sample questions
sample_questions = [
"What is the main purpose of this framework?",
"How do you install this tool?",
"What are the key components?",
"Explain the basic workflow",
"What are the best practices?"
]
return sample_questions |