diff --git a/config.json b/config.json index 49edee3..ff59bb6 100644 --- a/config.json +++ b/config.json @@ -1,3 +1,3 @@ { - "package_version_override": "0.1.1" + "package_version_override": "0.2.0" } \ No newline at end of file diff --git a/guardrails-api-client/guardrails_api_client/models/validate_payload_llm_api.py b/guardrails-api-client/guardrails_api_client/models/validate_payload_llm_api.py index 560092a..f4c203f 100644 --- a/guardrails-api-client/guardrails_api_client/models/validate_payload_llm_api.py +++ b/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" diff --git a/guardrails-api-client/setup.py b/guardrails-api-client/setup.py index 250d793..3fe8f87 100644 --- a/guardrails-api-client/setup.py +++ b/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", diff --git a/open-api-spec.yml b/open-api-spec.yml index d352366..db07370 100644 --- a/open-api-spec.yml +++ b/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 diff --git a/service-specs/guardrails-service-spec.yml b/service-specs/guardrails-service-spec.yml index 8abcb18..73ef2fb 100644 --- a/service-specs/guardrails-service-spec.yml +++ b/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: {}