** | 運(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'>