Spaces:
Paused
Paused
File size: 2,288 Bytes
ad33df7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from kotaemon.contribs.promptui.config import export_pipeline_to_config
from kotaemon.contribs.promptui.export import export_from_dict
from kotaemon.contribs.promptui.ui import build_from_dict
from .simple_pipeline import Pipeline
class TestPromptConfig:
def test_export_prompt_config(self):
"""Test if the prompt config is exported correctly"""
pipeline = Pipeline()
config_dict = export_pipeline_to_config(pipeline)
config = list(config_dict.values())[0]
assert "inputs" in config, "inputs should be in config"
assert "text" in config["inputs"], "inputs should have config"
assert "params" in config, "params should be in config"
assert "llm.deployment_name" in config["params"]
assert "llm.azure_endpoint" in config["params"]
assert "llm.openai_api_key" in config["params"]
assert "llm.openai_api_version" in config["params"]
assert "llm.request_timeout" in config["params"]
assert "llm.temperature" in config["params"]
class TestPromptUI:
def test_uigeneration(self):
"""Test if the gradio UI is exposed without any problem"""
pipeline = Pipeline()
config = export_pipeline_to_config(pipeline)
build_from_dict(config)
class TestExport:
def test_export(self, tmp_path):
"""Test if the export functionality works without error"""
from pathlib import Path
import yaml
from theflow.storage import storage
config_path = tmp_path / "config.yaml"
pipeline = Pipeline()
Path(storage.url(pipeline.config.store_result)).mkdir(
parents=True, exist_ok=True
)
config_dict = export_pipeline_to_config(pipeline)
pipeline_name = list(config_dict.keys())[0]
config_dict[pipeline_name]["logs"] = {
"sheet1": {
"inputs": [{"name": "text", "step": ".", "variable": "text"}],
"outputs": [{"name": "answer", "step": "."}],
},
}
with open(config_path, "w") as f:
yaml.safe_dump(config_dict, f)
export_from_dict(
config=str(config_path),
pipeline=pipeline_name,
output_path=str(tmp_path / "exported.xlsx"),
)
|