added spec

shohey1226 · Jul 2, 2023 · 68c67a9 · 68c67a9
1 parent e8ba82d
commit 68c67a9
Show file tree

Hide file tree

Showing 3 changed files with 215 additions and 1 deletion.
diff --git a/lib/llm_memory/broca.rb b/lib/llm_memory/broca.rb
@@ -48,7 +48,7 @@ def respond_with_schema(context: {}, schema: {})
       begin
         response = client.chat(
           parameters: {
-            model: @model,
+            model: "gpt-3.5-turbo-0613", # as of July 3, 2023
             messages: [
               {
                 role: "user",

diff --git a/...ixtures/vcr_cassettes/LlmMemory_Broca/generate_prompt/runs_respond_with_schema_method.yml b/...ixtures/vcr_cassettes/LlmMemory_Broca/generate_prompt/runs_respond_with_schema_method.yml
diff --git a/spec/llm_memory/broca_spec.rb b/spec/llm_memory/broca_spec.rb
@@ -47,5 +47,24 @@
       res = broca.respond(related_docs: related_docs, query_str: "what is my name?")
       expect(res).to include("Shohei")
     end
+
+    it "runs respond_with_schema method", :vcr do
+      related_docs = [{content: "My name is Shohei"}, {content: "I'm a software engineer"}]
+      broca = LlmMemory::Broca.new(prompt: template)
+      res = broca.respond_with_schema(
+        context: {related_docs: related_docs, query_str: "what is my name?"},
+        schema: {
+          type: :object,
+          properties: {
+            name: {
+              type: :string,
+              description: "The name of person"
+            }
+          },
+          required: ["name"]
+        }
+      )
+      expect(res).to include({name: "Shohei"})
+    end
   end
 end