Merge pull request #526 from guardrails-ai/status-fix

Fix Status

Author: CalebCourier

.github/workflows/ci.yml

@@ -173,7 +173,7 @@ jobs:
         - name: Install dependencies
           run: |
             make full;
-            poetry add openai==0.28.1 jupyter nbconvert cohere;
+            poetry add "openai>=1.2.4" jupyter nbconvert cohere;
         - name: Check for pypdfium2
           run: poetry run pip show pypdfium2
         - name: Huggingface Hub Login

.github/workflows/examples_check.yml

@@ -38,7 +38,7 @@ jobs:
     - name: Install dependencies
       run: |
         make full;
-        poetry add openai==0.28.1 jupyter nbconvert cohere;
+        poetry add "openai>=1.2.4" jupyter nbconvert cohere;
     - name: Check for pypdfium2
       run: poetry run pip show pypdfium2
     - name: Huggingface Hub Login
@@ -48,7 +48,7 @@ jobs:
         mkdir /tmp/nltk_data;
         poetry run python -m nltk.downloader -d /tmp/nltk_data punkt;
     - name: Use venv
-      run: source .venv/bin/activate
+      run: source $VENV
     - name: Execute notebooks and check for errors
       run: bash ./.github/workflows/scripts/run_notebooks.sh ${{ matrix.notebook }}
 

docs/examples/extracting_entities.ipynb

@@ -342,7 +342,7 @@
    "source": [
     "import openai\n",
     "\n",
-    "validated_response = guard(\n",
+    "_, validated_response, *rest = guard(\n",
     "    openai.chat.completions.create,\n",
     "    prompt_params={\"document\": content[:6000]},\n",
     "    max_tokens=2048,\n",
@@ -1724,7 +1724,7 @@
     }
    ],
    "source": [
-    "guard.state.most_recent_call.tree"
+    "guard.history.last.tree"
    ]
   }
  ],

docs/examples/generate_structured_data.ipynb

@@ -279,7 +279,7 @@
     "import openai\n",
     "\n",
     "\n",
-    "validated_output = guard(\n",
+    "_, validated_output, *rest = guard(\n",
     "    openai.chat.completions.create,\n",
     "    max_tokens=2048,\n",
     "    temperature=0\n",

docs/examples/guardrails_with_chat_models.ipynb

@@ -1712,7 +1712,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.17"
+   "version": "3.11.6"
   },
   "orig_nbformat": 4
  },

docs/examples/input_validation.ipynb

@@ -136,7 +136,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.17"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,

docs/examples/no_secrets_in_generated_text.ipynb

@@ -293,6 +293,19 @@
     "We can see that the output is a dictionary with the correct schema and types."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import openai\n",
+    "\n",
+    "raw_llm_response, validated_response, *rest = guard(\n",
+    "   openai.completions.create, model=\"text-davinci-003\", max_tokens=2048, temperature=0\n",
+    ")"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 9,

docs/examples/response_is_on_topic.ipynb

@@ -279,7 +279,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.17"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,

docs/examples/select_choice_based_on_action.ipynb

@@ -730,7 +730,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.17"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,

docs/examples/text_summarization_quality.ipynb

@@ -344,10 +344,10 @@
    "source": [
     "import openai\n",
     "\n",
-    "raw_llm_response, validated_response = guard(\n",
-    "    openai.Completion.create,\n",
+    "raw_llm_response, validated_response, *rest = guard(\n",
+    "    openai.completions.create,\n",
     "    prompt_params={'document': document},\n",
-    "    engine='text-davinci-003',\n",
+    "    model='text-davinci-003',\n",
     "    max_tokens=2048,\n",
     "    temperature=0\n",
     ")\n",
@@ -565,7 +565,7 @@
     }
    ],
    "source": [
-    "print(guard.state.most_recent_call.tree)"
+    "print(guard.history.last.tree)"
    ]
   },
   {
@@ -598,10 +598,10 @@
     }
    ],
    "source": [
-    "raw_llm_response, validated_response = guard(\n",
-    "    openai.Completion.create,\n",
+    "raw_llm_response, validated_response, *rest = guard(\n",
+    "    openai.completions.create,\n",
     "    prompt_params={'document': open(\"data/article1.txt\", \"r\").read()},\n",
-    "    engine='text-ada-001',\n",
+    "    model='text-ada-001',\n",
     "    max_tokens=512,\n",
     "    temperature=0\n",
     ")\n",
@@ -927,7 +927,7 @@
     }
    ],
    "source": [
-    "print(guard.state.most_recent_call.tree)"
+    "print(guard.history.last.tree)"
    ]
   }
  ],