29.RAG開(kāi)發(fā)-16-[擴(kuò)展]或運(yùn)算符的重寫(xiě)

** | 運(yùn)算符重載**

前文代碼中:chain = chat_prompt_template | model
在語(yǔ)法上使用了 | 運(yùn)算符的重寫(xiě)

在Python中,運(yùn)算符(如 +、I)的行為由類(lèi)的魔法方法決定。例如:

  • a + b本質(zhì)調(diào)用的是 a._ _ add _ _(b)
  • a | b本質(zhì)調(diào)用的是 a._ _ or _ _(b)

只需要自行實(shí)現(xiàn)類(lèi)的 _ _ or _ _ 方法,即可對(duì) | 符號(hào)的功能進(jìn)行重寫(xiě)。

示例:

  • 讓 a l b l c 的代碼得到一個(gè)自定義的類(lèi)對(duì)象(類(lèi)似列表即[a,b,c])
  • 調(diào)用run方法依次輸出 a、b、c
  • 我們需要重寫(xiě) | 即 _ _or _ _ 方法
# 序列管理類(lèi),用于存儲(chǔ)和執(zhí)行一系列對(duì)象
class MySequence(object):
    def __init__(self, *args):
        """初始化方法,接收任意數(shù)量的參數(shù)并存儲(chǔ)到序列中"""
        self.sequence = []  # 初始化一個(gè)空列表用于存儲(chǔ)序列元素
        for arg in args:
            self.sequence.append(arg)  # 將傳入的參數(shù)添加到序列中
            
    def __or__(self, other):
        """重寫(xiě)或運(yùn)算符(|),用于向序列中添加新元素"""
        self.sequence.append(other)  # 將新元素添加到序列中
        return self  # 返回自身,支持鏈?zhǔn)讲僮?    
    def run(self):
        """執(zhí)行序列中的每個(gè)元素,打印其字符串表示"""
        for arg in self.sequence:
            print(arg)  # 打印每個(gè)元素

# 測(cè)試類(lèi),用于演示或運(yùn)算符的重寫(xiě)
class Test(object):
    def __init__(self, name):
        """初始化方法,設(shè)置對(duì)象的名稱(chēng)"""
        self.name = name  # 存儲(chǔ)名稱(chēng)屬性
    
    def __str__(self):
        """重寫(xiě)字符串表示方法,返回格式化的字符串"""
        return f"Test({self.name})"  # 返回形如Test(名稱(chēng))的字符串
        
    def __or__(self, other):
        """重寫(xiě)或運(yùn)算符(|),創(chuàng)建并返回一個(gè)包含自身和另一個(gè)對(duì)象的MySequence實(shí)例"""
        return MySequence(self, other)  # 創(chuàng)建并返回MySequence實(shí)例

if __name__ == "__main__":
    # 創(chuàng)建三個(gè)Test實(shí)例
    test1 = Test("張三")
    test2 = Test("李四")
    test3 = Test("王五")
    
    # 使用或運(yùn)算符連接三個(gè)Test實(shí)例,創(chuàng)建MySequence對(duì)象
    # 執(zhí)行過(guò)程:test1 | test2 首先創(chuàng)建MySequence(test1, test2),然后 | test3 向序列中添加test3
    d = test1 | test2 | test3
    
    # 執(zhí)行序列中的所有元素
    d.run()
    
    # 打印d的類(lèi)型,驗(yàn)證其為MySequence實(shí)例
    print(type(d))

Test(張三)
Test(李四)
Test(王五)
<class '__main__.MySequence'>
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容