Chain
Chain的 or 原理:
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
class Test(object):
def __init__(self, name):
self.name = name
def __or__(self, other):
# 把自己跟other 作為參數傳給物件
# a | b 參數就是a,b
# a | b 傳回的是 MySequence 物件
return MySequence(self, other)
def __str__(self):
return self.name
class MySequence(object):
# a | b 傳回的是 MySequence 物件
# *args 為 a , b
def __init__(self, *args):
self.sequence = []
for arg in args:
print(f"init arg: {arg}")
self.sequence.append(arg)
# a | b | c
# 因為 a | b 傳回的是 MySequence 物件，與 c 進行 or 的時候，使用的是MySequence的 or 方法
def __or__(self, other):
# 會輸出 c
print(f"or other: {other}")
self.sequence.append(other)
return self
def run(self):
for i in self.sequence:
print(i)
if __name__ == '__main__':
a = Test('a')
b = Test('b')
c = Test('c')
d = a | b | c
d.run()
init arg: a
init arg: b
or other: c
a
b
c
繼承關係
Runnable 祖先類別
↑
RunnableSerializable 祖父類別
↑
BasePromptTemplate 父類別
↑
PromptTemplate 子類別
Runnable 祖先類別中，有一個__or__() 方法，會傳回RunnableSequence 物件。
RunnableSequence 物件也有__or__() 方法。
代入上面的or範例，Runnable 類別等同於Test 類別。
RunnableSequence 類別，等同 MySequence 類別。
chain的類別為 RunnableSequence
RunnableSequence 裡面有 __or__ 函式，代表可以使用 | 也就是 __or__ 函式，使用 a | b | c | d的方式，把prompt 作為參數，傳給 llm。
1
2
3
4
5
6
7
from langchain_core.prompts import PromptTemplate
from langchain_google_genai import ChatGoogleGenerativeAI
prompt = PromptTemplate.from_template("你是一個AI助手")
llm = ChatGoogleGenerativeAI(model="gemini-3-flash-preview", temperature=0)
chain = prompt | llm
print(type(chain))
<class 'langchain_core.runnables.base.RunnableSequence'>