16 lines
652 B
Python
16 lines
652 B
Python
""" Content to Propositions"""
|
|
from typing import Type
|
|
from pydantic import BaseModel
|
|
from cognitive_architecture.infrastructure.llm.get_llm_client import get_llm_client
|
|
from cognitive_architecture.shared.data_models import KnowledgeGraph
|
|
|
|
async def generate_graph(text_input:str,system_prompt_path:str, response_model: Type[BaseModel]):
|
|
doc_path = "cognitive_architecture/infrastructure/llm/prompts/generate_graph_prompt.txt"
|
|
llm_client = get_llm_client()
|
|
|
|
return await llm_client.generate_graph(text_input,system_prompt_path, response_model)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
generate_graph("test", "test", response_model=KnowledgeGraph)
|
|
|