support store for openai (#3399)

This commit is contained in:
Srishti Gureja
2025-09-02 21:23:59 +05:30
committed by GitHub
parent 64cbe84089
commit cc15a22bf9
3 changed files with 12 additions and 3 deletions
+3
View File
@@ -28,6 +28,7 @@ class OpenAIConfig(BaseLlmConfig):
openrouter_base_url: Optional[str] = None,
site_url: Optional[str] = None,
app_name: Optional[str] = None,
store: bool = False,
# Response monitoring callback
response_callback: Optional[Callable[[Any, dict, dict], None]] = None,
):
@@ -72,5 +73,7 @@ class OpenAIConfig(BaseLlmConfig):
self.openrouter_base_url = openrouter_base_url
self.site_url = site_url
self.app_name = app_name
self.store = store
# Response monitoring
self.response_callback = response_callback
+7 -1
View File
@@ -123,7 +123,13 @@ class OpenAILLM(LLMBase):
openrouter_params["extra_headers"] = extra_headers
params.update(**openrouter_params)
else:
openai_specific_generation_params = ["store"]
for param in openai_specific_generation_params:
if hasattr(self.config, param):
params[param] = getattr(self.config, param)
if response_format:
params["response_format"] = response_format
if tools: # TODO: Remove tools if no issues found with new memory addition logic