Skip to content

Commit b493e2e

Browse files
committed
feat: add azure openai inference
New inference provider. Relates to: RHAIENG-744 Signed-off-by: Sébastien Han <seb@redhat.com>
1 parent a27e6e5 commit b493e2e

File tree

4 files changed

+10
-0
lines changed

4 files changed

+10
-0
lines changed

distribution/Containerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ RUN pip install \
1616
fire \
1717
httpx \
1818
ibm_watsonx_ai \
19+
litellm \
1920
matplotlib \
2021
mcp>=1.8.1 \
2122
nltk \

distribution/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ You can see an overview of the APIs and Providers the image ships with in the ta
1414
| eval | remote::trustyai_lmeval |
1515
| files | inline::localfs |
1616
| inference | inline::sentence-transformers |
17+
| inference | remote::azure |
1718
| inference | remote::bedrock |
1819
| inference | remote::vllm |
1920
| inference | remote::watsonx |

distribution/build.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ distribution_spec:
77
- provider_type: remote::bedrock
88
- provider_type: inline::sentence-transformers
99
- provider_type: remote::watsonx
10+
- provider_type: remote::azure
1011
vector_io:
1112
- provider_type: inline::milvus
1213
- provider_type: remote::milvus

distribution/run.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,13 @@ providers:
4242
url: ${env.WATSONX_BASE_URL:=https://us-south.ml.cloud.ibm.com}
4343
api_key: ${env.WATSONX_API_KEY:=}
4444
project_id: ${env.WATSONX_PROJECT_ID:=}
45+
- provider_id: ${env.AZURE_API_KEY:+azure}
46+
provider_type: remote::azure
47+
config:
48+
api_key: ${env.AZURE_API_KEY:=}
49+
api_base: ${env.AZURE_API_BASE:=}
50+
api_version: ${env.AZURE_API_VERSION:=}
51+
api_type: ${env.AZURE_API_TYPE:=}
4552
vector_io:
4653
- provider_id: milvus
4754
provider_type: inline::milvus

0 commit comments

Comments
 (0)