LangChain openAI快速接入。

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from langchain.llms import OpenAI
+
+def chatOpenAI(text):
+    llm = OpenAI(temperature=0.9)
+    return llm(text)
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        '''
+        # LangChain Test
+        LLM先跑起来，GO GO。
+        LangChain接入openAI，简单测试。
+        ''')
+    gr.Interface(fn=chatOpenAI, inputs="text", outputs="text")
+demo.launch()

openai_agent.py

@@ -0,0 +1,17 @@
+from langchain.agents import load_tools
+from langchain.agents import initialize_agent
+from langchain.agents import AgentType
+from langchain.llms import OpenAI
+
+# First, let's load the language model we're going to use to control the agent.
+llm = OpenAI(temperature=0)
+
+# Next, let's load some tools to use. Note that the `llm-math` tool uses an LLM, so we need to pass that in.
+tools = load_tools(["serpapi", "llm-math"], llm=llm)
+
+
+# Finally, let's initialize an agent with the tools, the language model, and the type of agent we want to use.
+agent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
+
+# Now let's test it out!
+agent.run("昨天北京气温如何?")

openai_prompt_template.py

@@ -0,0 +1,11 @@
+from langchain.prompts import PromptTemplate
+from langchain.llms import OpenAI
+from langchain.chains import LLMChain
+
+llm = OpenAI(temperature=0.9)
+prompt = PromptTemplate(
+    input_variables=["product"],
+    template="What is a good name for a company that makes {product}?"
+)
+chain = LLMChain(llm=llm,prompt=prompt)
+print(chain.run("colorful socks"))

openai_simple.py

@@ -0,0 +1,9 @@
+#!/usr/bin/python3
+
+from langchain.llms import OpenAI
+import os
+
+llm = OpenAI(temperature=0.9)
+
+text = "Hi, Could you help me choose a Chinese name? For example"
+print(llm(text))

socket_client.py

@@ -0,0 +1,15 @@
+#!/usr/bin/python3
+import socket
+import sys
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+host = socket.gethostname()
+port = 9999
+s.connect((host,port))
+while True:
+    msg = s.recv(1024).decode("utf-8")
+    print("server:",msg)
+    msg = input("client: ")
+    if not msg: break
+    s.sendall(msg.encode("utf-8"))
+s.close()

socket_server.py

@@ -0,0 +1,22 @@
+#!/usr/bin/python3
+import socket
+import sys
+
+serverSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+host = socket.gethostname()
+port = 9999
+serverSocket.bind((host, port))
+serverSocket.listen(5)
+
+while True:
+    clientSocket,addr = serverSocket.accept()
+    print("已连接地址：%s"%str(addr))
+
+    msg="hello , I'm server.\n"
+    clientSocket.send(msg.encode('utf-8'))
+    while True:
+        data = clientSocket.recv(1024).decode('utf-8')
+        print ("client:%s"% data)
+        inputMsg = input("server：")
+        clientSocket.send(inputMsg.encode('utf-8'))
+clientSocket.close()