WenxinWorkshop Python SDK

• A third-party Python SDK for a WenxinWorkshop.

Quick Start

• Install wenxinworkshop SDK

  $ pip install git+https://github.com/jm12138/WenxinWorkshop-Python-SDK

• Import wenxinworkshop SDK

  from wenxinworkshop import LLMAPI, EmbeddingAPI, PromptTemplateAPI
  from wenxinworkshop import Message, Messages, Texts

• Set API key and Secret key

  api_key = '...'
  secret_key = '...'

• LLM chat

  # create a LLM API
  erniebot = LLMAPI(
      api_key=api_key,
      secret_key=secret_key,
      url=LLMAPI.ERNIEBot
  )
  
  # create a message
  message = Message(
      role='user',
      content='你好！'
  )
  
  # create messages
  messages: Messages = [message]
  
  # get response from LLM API
  response = erniebot(
      messages=messages,
      temperature=None,
      top_p=None,
      penalty_score=None,
      stream=None,
      user_id=None,
      chunk_size=512
  )
  
  # print response
  print(response)
  
  # get response stream from LLM API
  response_stream = erniebot(
      messages=messages,
      temperature=None,
      top_p=None,
      penalty_score=None,
      stream=True,
      user_id=None,
      chunk_size=512
  )
  
  # print response stream
  for item in response_stream:
      print(item, end='')

• Embedding

  # create a Embedding API
  ernieembedding = EmbeddingAPI(
      api_key=api_key,
      secret_key=secret_key,
      url=EmbeddingAPI.EmbeddingV1
  )
  
  # create texts
  texts: Texts = [
      '你好！',
      '你好吗？',
      '你是谁？'
  ]
  
  # get embeddings from Embedding API
  response = ernieembedding(
      texts=texts,
      user_id=None
  )
  
  # print embeddings
  print(response)

• Get prompt template

  # create a Prompt Template API
  prompttemplate = PromptTemplateAPI(
      api_key=api_key,
      secret_key=secret_key,
      url=PromptTemplateAPI.PromptTemplate
  )
  
  # get prompt template from Prompt Template API
  response = prompttemplate(
      template_id=1968,
      content='侏罗纪世界'
  )
  
  # print prompt template
  print(response)