feat: add citation style quick setting (#554) bump:patch

2024-12-05 19:55:59 +07:00
parent 7d8f40e841
commit 6650b15c64
6 changed files with 46 additions and 13 deletions
--- a/libs/kotaemon/kotaemon/llms/chats/langchain_based.py
+++ b/libs/kotaemon/kotaemon/llms/chats/langchain_based.py
@@ -296,6 +296,15 @@ class LCGeminiChat(LCChatMixin, ChatLLM):  # type: ignore
        required=True,
    )

+    def _get_tool_call_kwargs(self):
+        return {
+            "tool_config": {
+                "function_calling_config": {
+                    "mode": "ANY",
+                }
+            }
+        }
+
    def __init__(
        self,
        api_key: str | None = None,