소스 검색

feat: pass QueryConfig to dry_run (#173)

gasolin 2 년 전
부모
커밋
5e6aef1e37
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      embedchain/embedchain.py

+ 2 - 2
embedchain/embedchain.py

@@ -279,7 +279,7 @@ class EmbedChain:
         memory.chat_memory.add_ai_message(answer)
         return answer
 
-    def dry_run(self, input_query):
+    def dry_run(self, input_query, config: QueryConfig = None):
         """
         A dry run does everything except send the resulting prompt to
         the LLM. The purpose is to test the prompt, not the response.
@@ -387,4 +387,4 @@ class OpenSourceApp(EmbedChain):
         response = gpt4all_model.generate(
             prompt=prompt,
         )
-        return response
+        return response