Open lucafirefox opened 6 months ago
That seems more than reasonable. Makes you wonder why I didn't do that when I created this... 😅 That said, if you want to work on this, that would be great! If not, I'll put it on my (very long) to-do list.
No problem, PR done. Thanks for the support.
I am using keybert for a topic extraction task and I have the need of making the system content as a variable so I can change accordingly based on my task. Here below I represent what I mean.
Current implementation:
Proposed implementation:
It implies replacing the hard coded string "You are a helpful assistant." with a variable coming from init which default value is actually the existing content. Retrocompatibility is guaranteed. This logic can be applied both on
keybert/llm/_openai.py
andkeybert/llm/_litellm.py
files.