Skip to content

Arize phoenix query engine

ArizePhoenixQueryEnginePack #

Bases: BaseLlamaPack

Arize-Phoenix LlamaPack 展示了如何使用跟踪工具来对 LlamaIndex 查询引擎进行仪器化。它在后台启动 Phoenix,构建一个输入节点列表上的索引,并实例化和仪器化一个查询引擎,以便将每个查询的跟踪数据发送到 Phoenix。

注意:使用此 LlamaPack 需要通过 OPENAI_API_KEY 环境变量设置您的 OpenAI API 密钥。

Source code in llama_index/packs/arize_phoenix_query_engine/base.py
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
class ArizePhoenixQueryEnginePack(BaseLlamaPack):
    """Arize-Phoenix LlamaPack 展示了如何使用跟踪工具来对 LlamaIndex 查询引擎进行仪器化。它在后台启动 Phoenix,构建一个输入节点列表上的索引,并实例化和仪器化一个查询引擎,以便将每个查询的跟踪数据发送到 Phoenix。

注意:使用此 LlamaPack 需要通过 OPENAI_API_KEY 环境变量设置您的 OpenAI API 密钥。"""

    def __init__(
        self,
        nodes: List[TextNode],
        **kwargs: Any,
    ) -> None:
        """初始化一个新的ArizePhoenixQueryEnginePack实例。

Args:
    nodes(List[TextNode]):用于构建索引的节点输入列表。
"""
        try:
            import phoenix as px
        except ImportError:
            raise ImportError(
                "The arize-phoenix package could not be found. "
                "Please install with `pip install arize-phoenix`."
            )
        self._session: "PhoenixSession" = px.launch_app()
        set_global_handler("arize_phoenix")
        self._index = VectorStoreIndex(nodes, **kwargs)
        self._query_engine = self._index.as_query_engine()

    def get_modules(self) -> Dict[str, Any]:
        """返回一个包含LlamaPack内部信息的字典。

返回:
    Dict[str, Any]: 包含LlamaPack内部信息的字典。
"""
        return {
            "session": self._session,
            "session_url": self._session.url,
            "index": self._index,
            "query_engine": self._query_engine,
        }

    def run(self, *args: Any, **kwargs: Any) -> Any:
        """对索引运行查询。

返回:
    任何:来自查询引擎的响应。
"""
        return self._query_engine.query(*args, **kwargs)

get_modules #

get_modules() -> Dict[str, Any]

返回一个包含LlamaPack内部信息的字典。

返回: Dict[str, Any]: 包含LlamaPack内部信息的字典。

Source code in llama_index/packs/arize_phoenix_query_engine/base.py
43
44
45
46
47
48
49
50
51
52
53
54
    def get_modules(self) -> Dict[str, Any]:
        """返回一个包含LlamaPack内部信息的字典。

返回:
    Dict[str, Any]: 包含LlamaPack内部信息的字典。
"""
        return {
            "session": self._session,
            "session_url": self._session.url,
            "index": self._index,
            "query_engine": self._query_engine,
        }

run #

run(*args: Any, **kwargs: Any) -> Any

对索引运行查询。

返回: 任何:来自查询引擎的响应。

Source code in llama_index/packs/arize_phoenix_query_engine/base.py
56
57
58
59
60
61
62
    def run(self, *args: Any, **kwargs: Any) -> Any:
        """对索引运行查询。

返回:
    任何:来自查询引擎的响应。
"""
        return self._query_engine.query(*args, **kwargs)