Update llm/llm.py
Browse files- llm/llm.py +1 -0
llm/llm.py
CHANGED
|
@@ -29,6 +29,7 @@ class LLM_chain:
|
|
| 29 |
def __call__(self, entity: str, id: int = 0):
|
| 30 |
try:
|
| 31 |
data = self.__read_yaml()
|
|
|
|
| 32 |
prompts = data["prompts"]
|
| 33 |
template = prompts["prompt_template"][1]
|
| 34 |
prompt = PromptTemplate(template=template, input_variables=["entity"])
|
|
|
|
| 29 |
def __call__(self, entity: str, id: int = 0):
|
| 30 |
try:
|
| 31 |
data = self.__read_yaml()
|
| 32 |
+
print(data)
|
| 33 |
prompts = data["prompts"]
|
| 34 |
template = prompts["prompt_template"][1]
|
| 35 |
prompt = PromptTemplate(template=template, input_variables=["entity"])
|