module nemoguardrails.rails.llm.config#

用于 rails 配置的模块。


class Model#

由 rails 引擎使用的模型的配置。

通常,配置主模型,例如:{ “type”: “main”, “engine”: “openai”, “model”: “gpt-3.5-turbo-instruct” }


class Instruction#

用于自然语言指令的配置,应传递给 LLM。


class Document#

用于问答的文档的配置。


class SensitiveDataDetectionOptions#


class SensitiveDataDetection#

应检测哪些敏感数据的配置。


class MessageTemplate#

消息结构的模板。


class TaskPrompt#

将用于特定任务的提示的配置。


classmethod TaskPrompt.check_fields#

check_fields(values)

class EmbeddingSearchProvider#

嵌入搜索提供程序的配置。


class KnowledgeBaseConfig#


class CoreConfig#

核心内部机制的设置。


class InputRails#

输入 rails 的配置。


class OutputRails#

输出 rails 的配置。


class RetrievalRails#

检索 rails 的配置。


class SingleCallConfig#

主题 rails 的单 LLM 调用选项的配置。


class UserMessagesConfig#

关于如何解释用户消息的配置。


class DialogRails#

主题 rails 的配置。


class FactCheckingRailConfig#

用于事实检查 rail 的配置数据。


class RailsConfigData#

用于开箱即用的特定 rails 的配置数据。


class Rails#

特定 rails 的配置。


class RailsConfig#

用于模型和 rails 的配置对象。

TODO:为 user_messages、bot_messages 和 flows 添加类型化配置。


property RailsConfig.streaming_supported#

当前配置是否支持流式传输。

目前,如果存在输出 rails,则不支持流式传输。


method RailsConfig.from_content#

from_content(
    colang_content: Optional[str] = None,
    yaml_content: Optional[str] = None,
    config: Optional[dict] = None
)

从提供的 colang/YAML 内容/配置字典加载配置。


method RailsConfig.from_path#

from_path(
    config_path: str,
    test_set_percentage: Optional[float] = 0.0,
    test_set: Optional[Dict[str, List]] = {},
    max_samples_per_intent: Optional[int] = 0
)

从给定路径加载配置。

支持从单个文件或目录加载。

也用于测试 Guardrails 应用程序,在这种情况下,测试集是从配置文件中的意图样本随机创建的。在这种情况下,test_set_percentage 应大于 0。

如果我们想限制每个意图的样本数量,请将 max_samples_per_intent 设置为正数。 它对于测试应用程序很有用,但也用于在某些情况下限制意图中样本的数量。 对于每个意图,选择的样本是随机选择的。


classmethod RailsConfig.parse_object#

parse_object(obj)

从给定的字典解析配置对象。