Hugging Face's logo

Spaces:
Shipmaster1
/
PythonicRAGReact
Runtime error

App Files Community
PythonicRAGReact
File size: 733 Bytes
d184dfc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
from typing import Dict, Any


class SystemRolePrompt:
    def __init__(self, template: str):
        self.template = template

    def create_message(self, **kwargs) -> Dict[str, str]:
        return {"role": "system", "content": self.template.format(**kwargs)}


class UserRolePrompt:
    def __init__(self, template: str):
        self.template = template

    def create_message(self, **kwargs) -> Dict[str, str]:
        return {"role": "user", "content": self.template.format(**kwargs)}


class AssistantRolePrompt:
    def __init__(self, template: str):
        self.template = template

    def create_message(self, **kwargs) -> Dict[str, str]:
        return {"role": "assistant", "content": self.template.format(**kwargs)}