Skip to content
This repository was archived by the owner on Feb 6, 2026. It is now read-only.

entry #20

Merged
denverbaumgartner merged 4 commits into from
Mar 10, 2025
Merged

entry #20

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
40382ce
refactor(config): break out mlflow variables in configs
denverbaumgartner Mar 10, 2025
df052b9
refactor: break out graphdoc class to config module
denverbaumgartner Mar 10, 2025
7092778
test: update test to not utilize graphdoc
denverbaumgartner Mar 10, 2025
b99f246
build: update runners to not utilize graphdoc
denverbaumgartner Mar 10, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions graphdoc/assets/configs/single_prompt_doc_generator_module.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ graphdoc:
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

mlflow:
mlflow_tracking_uri: !env MLFLOW_TRACKING_URI # The tracking URI for MLflow
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

language_model:
model: openai/gpt-4o # Must be a valid dspy language model
api_key: !env OPENAI_API_KEY # Must be a valid dspy language model API key
Expand Down
5 changes: 5 additions & 0 deletions graphdoc/assets/configs/single_prompt_doc_generator_module_eval.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ graphdoc:
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

mlflow:
mlflow_tracking_uri: !env MLFLOW_TRACKING_URI # The tracking URI for MLflow
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

language_model:
model: openai/gpt-4o # Must be a valid dspy language model
api_key: !env OPENAI_API_KEY # Must be a valid dspy language model API key
Expand Down
5 changes: 5 additions & 0 deletions graphdoc/assets/configs/single_prompt_doc_generator_trainer.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ graphdoc:
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

mlflow:
mlflow_tracking_uri: !env MLFLOW_TRACKING_URI # The tracking URI for MLflow
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

language_model:
model: openai/gpt-4o # Must be a valid dspy language model
api_key: !env OPENAI_API_KEY # Must be a valid dspy language model API key
Expand Down
5 changes: 5 additions & 0 deletions graphdoc/assets/configs/single_prompt_doc_quality_trainer.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ graphdoc:
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

mlflow:
mlflow_tracking_uri: !env MLFLOW_TRACKING_URI # The tracking URI for MLflow
mlflow_tracking_username: !env MLFLOW_TRACKING_USERNAME # The username for the mlflow tracking server
mlflow_tracking_password: !env MLFLOW_TRACKING_PASSWORD # The password for the mlflow tracking server

language_model:
model: openai/gpt-4o # Must be a valid dspy language model
api_key: !env OPENAI_API_KEY # Must be a valid dspy language model API key
Expand Down
28 changes: 28 additions & 0 deletions graphdoc/graphdoc/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,21 @@
# Copyright 2025-, Semiotic AI, Inc.
# SPDX-License-Identifier: Apache-2.0

from graphdoc.config import (
doc_generator_eval_from_yaml,
doc_generator_module_from_dict,
doc_generator_module_from_yaml,
mlflow_data_helper_from_dict,
mlflow_data_helper_from_yaml,
single_prompt_from_dict,
single_prompt_from_yaml,
single_trainer_from_dict,
single_trainer_from_yaml,
split_trainset,
trainset_and_evalset_from_yaml,
trainset_from_dict,
trainset_from_yaml,
)
from graphdoc.data import (
DspyDataHelper,
GenerationDataHelper,
Expand Down Expand Up @@ -79,4 +94,17 @@
"load_yaml_config",
"schema_objects_to_dataset",
"setup_logging",
"mlflow_data_helper_from_dict",
"mlflow_data_helper_from_yaml",
"trainset_from_dict",
"trainset_from_yaml",
"split_trainset",
"trainset_and_evalset_from_yaml",
"single_prompt_from_dict",
"single_prompt_from_yaml",
"single_trainer_from_dict",
"single_trainer_from_yaml",
"doc_generator_module_from_dict",
"doc_generator_module_from_yaml",
"doc_generator_eval_from_yaml",
]
Loading