30 lines
783 B
Python
30 lines
783 B
Python
from dataclasses import dataclass
|
|
from typing import Optional, List
|
|
|
|
@dataclass
|
|
class GraphRAGResponse:
|
|
|
|
question: str
|
|
|
|
answer: str
|
|
|
|
evidence: Optional[str] = None
|
|
|
|
entities_used: Optional[List[str]] = None
|
|
|
|
relations_used: Optional[List[str]] = None
|
|
|
|
source_documents: Optional[List[str]] = None
|
|
|
|
def __str__(self):
|
|
|
|
return (
|
|
"\nGraphRAGResponse(\n"
|
|
f" question = {self.question!r}\n"
|
|
f" answer = {self.answer!r}\n"
|
|
f" evidence = {self.evidence!r}\n"
|
|
f" entities_used = {self.entities_used!r}\n"
|
|
f" relations_used = {self.relations_used!r}\n"
|
|
f" source_documents = {self.source_documents!r}\n"
|
|
")"
|
|
) |