Upgrade the declarative pipeline for cleaner interface (#51)

2023-10-24 11:12:22 +07:00
parent aab982ddc4
commit 9035e25666
26 changed files with 365 additions and 169 deletions
--- a/knowledgehub/llms/completions/base.py
+++ b/knowledgehub/llms/completions/base.py
@@ -27,7 +27,7 @@ class LangchainLLM(LLM):
                self._kwargs[param] = params.pop(param)
        super().__init__(**params)

-    @Param.decorate(no_cache=True)
+    @Param.auto(cache=False)
    def agent(self):
        return self._lc_class(**self._kwargs)