Skip to content

Commit 6c13cce

Browse files
authored
docs: update configuration docs (#1155)
* update configuration docs * update configuration docs * update configuration docs
1 parent ef749ab commit 6c13cce

File tree

5 files changed

+32
-5
lines changed

5 files changed

+32
-5
lines changed

README.md

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,9 @@ More details can be found in the [development setup](https://rdagent.readthedocs
172172
- **Using LiteLLM (Default)**: We now support LiteLLM as a backend for integration with multiple LLM providers. You can configure in multiple ways:
173173

174174
**Option 1: Unified API base for both models**
175+
176+
*Configuration Example: `OpenAI` Setup :*
177+
175178
```bash
176179
cat << EOF > .env
177180
# Set to any model supported by LiteLLM.
@@ -182,6 +185,19 @@ More details can be found in the [development setup](https://rdagent.readthedocs
182185
OPENAI_API_KEY=<replace_with_your_openai_api_key>
183186
```
184187
188+
*Configuration Example: `Azure OpenAI` Setup :*
189+
190+
> Before using this configuration, please confirm in advance that your `Azure OpenAI API key` supports `embedded models`.
191+
192+
```bash
193+
cat << EOF > .env
194+
EMBEDDING_MODEL=azure/<Model deployment supporting embedding>
195+
CHAT_MODEL=azure/<your deployment name>
196+
AZURE_API_KEY=<replace_with_your_openai_api_key>
197+
AZURE_API_BASE=<your_unified_api_base>
198+
AZURE_API_VERSION=<azure api version>
199+
```
200+
185201
**Option 2: Separate API bases for Chat and Embedding models**
186202
```bash
187203
cat << EOF > .env
@@ -201,7 +217,7 @@ More details can be found in the [development setup](https://rdagent.readthedocs
201217
LITELLM_PROXY_API_BASE=https://api.siliconflow.cn/v1
202218
```
203219
204-
**Configuration Example: DeepSeek Setup**:
220+
*Configuration Example: `DeepSeek` Setup :*
205221
206222
>Since many users encounter configuration errors when setting up DeepSeek. Here's a complete working example for DeepSeek Setup:
207223
```bash

docs/installation_and_configuration.rst

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,17 @@ To align with the Python SDK example above, you can configure the `CHAT_MODEL` b
148148
149149
This configuration allows you to call Azure OpenAI through LiteLLM while using an external provider (e.g., SiliconFlow) for embeddings.
150150

151+
If your `Azure OpenAI API Key`` supports `embedding model`, you can refer to the following configuration example.
152+
153+
.. code-block:: Properties
154+
155+
cat << EOF > .env
156+
EMBEDDING_MODEL=azure/<Model deployment supporting embedding>
157+
CHAT_MODEL=azure/<your deployment name>
158+
AZURE_API_KEY=<replace_with_your_openai_api_key>
159+
AZURE_API_BASE=<your_unified_api_base>
160+
AZURE_API_VERSION=<azure api version>
161+
151162
Configuration(deprecated)
152163
=========================
153164

rdagent/scenarios/data_science/example/arf-12-hours-prediction-task/sample.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def create_debug_data(
6666
min_frac: float = 0.02,
6767
min_num: int = 10,
6868
):
69-
dataset_root = Path(dataset_path) / "arf-12-hour-prediction-task"
69+
dataset_root = Path(dataset_path) / "arf-12-hours-prediction-task"
7070
output_root = Path(output_path)
7171

7272
for sub in ["train", "test"]:

rdagent/scenarios/data_science/example/eval/arf-12-hours-prediction-task/grade.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def grade(submission: pd.DataFrame, answers: pd.DataFrame) -> float:
6060
print(
6161
json.dumps(
6262
{
63-
"competition_id": "arf-12-hour-prediction-task",
63+
"competition_id": "arf-12-hours-prediction-task",
6464
"score": score,
6565
}
6666
)

rdagent/scenarios/data_science/example/source_data/arf-12-hours-prediction-task/prepare.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
raw_feature_path = CURRENT_DIR / "X.npz"
1212
raw_label_path = CURRENT_DIR / "ARF_12h.csv"
1313

14-
public = ROOT_DIR / "arf-12-hour-prediction-task"
15-
private = ROOT_DIR / "eval" / "arf-12-hour-prediction-task"
14+
public = ROOT_DIR / "arf-12-hours-prediction-task"
15+
private = ROOT_DIR / "eval" / "arf-12-hours-prediction-task"
1616

1717
if not (public / "test").exists():
1818
(public / "test").mkdir(parents=True, exist_ok=True)

0 commit comments

Comments
 (0)