Skip to content

Commit d636519

Browse files
committed
fixes Prompt handling
1 parent e878451 commit d636519

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

ngwidgets/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "0.20.3"
1+
__version__ = "0.20.4"

ngwidgets/llm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ def from_chat(self,
4747
result:str,
4848
chat_completion,
4949
start_time:datetime,
50+
model:Optional[str]=None,
5051
image_path:Optional[str]=None,
5152
temperature:float=0.7) -> 'Prompt':
5253
"""
@@ -58,7 +59,7 @@ def from_chat(self,
5859
prompt = Prompt(
5960
prompt=prompt_text,
6061
response=result,
61-
model=self.model,
62+
model=model,
6263
model_details=model_details,
6364
temperature=temperature,
6465
tokens=total_tokens,

0 commit comments

Comments
 (0)