Merge pull request #12 from guardrails-ai/litellm-support

add litellm completion and v1 openai methods

Author: CalebCourier

config.json

@@ -1,3 +1,3 @@
 {
-  "package_version_override": "0.1.1"
+  "package_version_override": "0.2.0"
 }

guardrails-api-client/guardrails_api_client/models/validate_payload_llm_api.py

@@ -2,8 +2,12 @@
 
 
 class ValidatePayloadLlmApi(str, Enum):
+    LITELLM_ACOMPLETION = "litellm.acompletion"
+    LITELLM_COMPLETION = "litellm.completion"
     OPENAI_CHATCOMPLETION_ACREATE = "openai.ChatCompletion.acreate"
     OPENAI_CHATCOMPLETION_CREATE = "openai.ChatCompletion.create"
+    OPENAI_CHAT_COMPLETIONS_CREATE = "openai.chat.completions.create"
+    OPENAI_COMPLETIONS_CREATE = "openai.completions.create"
     OPENAI_COMPLETION_ACREATE = "openai.Completion.acreate"
     OPENAI_COMPLETION_CREATE = "openai.Completion.create"
 

guardrails-api-client/setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name="guardrails-api-client",
-    version="0.1.1",
+    version="0.2.0",
     description="A client library for accessing Guardrails API",
     long_description=long_description,
     long_description_content_type="text/markdown",

open-api-spec.yml

@@ -780,7 +780,7 @@ components:
           additionalProperties: {}
         llmApi:
           type: string
-          enum: ["openai.Completion.create", "openai.ChatCompletion.create", "openai.Completion.acreate", "openai.ChatCompletion.acreate"]
+          enum: ["openai.Completion.create", "openai.completions.create", "openai.ChatCompletion.create", "openai.chat.completions.create", "openai.Completion.acreate", "openai.ChatCompletion.acreate", "litellm.completion", "litellm.acompletion"]
       additionalProperties: {}
     Ingestion:
       type: object

service-specs/guardrails-service-spec.yml

@@ -493,8 +493,12 @@ components:
           enum:
             [
               "openai.Completion.create",
+              "openai.completions.create",
               "openai.ChatCompletion.create",
+              "openai.chat.completions.create",
               "openai.Completion.acreate",
               "openai.ChatCompletion.acreate",
+              "litellm.completion",
+              "litellm.acompletion"
             ]
       additionalProperties: {}