mirror of
https://github.com/hwchase17/langchain.git
synced 2026-06-09 10:17:00 +00:00
style: .. code-block:: admonition translations (#33400)
biiiiiiiiiiiiiiiigggggggg pass
This commit is contained in:
@@ -319,15 +319,15 @@ class ChatHuggingFace(BaseChatModel):
|
||||
Install `langchain-huggingface` and ensure your Hugging Face token
|
||||
is saved.
|
||||
|
||||
.. code-block:: bash
|
||||
```bash
|
||||
pip install langchain-huggingface
|
||||
```
|
||||
|
||||
pip install langchain-huggingface
|
||||
```python
|
||||
from huggingface_hub import login
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
from huggingface_hub import login
|
||||
|
||||
login() # You will be prompted for your HF key, which will then be saved locally
|
||||
login() # You will be prompted for your HF key, which will then be saved locally
|
||||
```
|
||||
|
||||
Key init args — completion params:
|
||||
llm: `HuggingFaceTextGenInference`, `HuggingFaceEndpoint`, `HuggingFaceHub`, or
|
||||
@@ -348,129 +348,129 @@ class ChatHuggingFace(BaseChatModel):
|
||||
section.
|
||||
|
||||
Instantiate:
|
||||
.. code-block:: python
|
||||
```python
|
||||
from langchain_huggingface import HuggingFaceEndpoint,
|
||||
ChatHuggingFace
|
||||
|
||||
from langchain_huggingface import HuggingFaceEndpoint,
|
||||
ChatHuggingFace
|
||||
llm = HuggingFaceEndpoint(
|
||||
repo_id="microsoft/Phi-3-mini-4k-instruct",
|
||||
task="text-generation",
|
||||
max_new_tokens=512,
|
||||
do_sample=False,
|
||||
repetition_penalty=1.03,
|
||||
)
|
||||
|
||||
llm = HuggingFaceEndpoint(
|
||||
repo_id="microsoft/Phi-3-mini-4k-instruct",
|
||||
task="text-generation",
|
||||
max_new_tokens=512,
|
||||
do_sample=False,
|
||||
repetition_penalty=1.03,
|
||||
)
|
||||
|
||||
chat = ChatHuggingFace(llm=llm, verbose=True)
|
||||
chat = ChatHuggingFace(llm=llm, verbose=True)
|
||||
```
|
||||
|
||||
Invoke:
|
||||
.. code-block:: python
|
||||
```python
|
||||
messages = [
|
||||
("system", "You are a helpful translator. Translate the user
|
||||
sentence to French."),
|
||||
("human", "I love programming."),
|
||||
]
|
||||
|
||||
messages = [
|
||||
("system", "You are a helpful translator. Translate the user
|
||||
sentence to French."),
|
||||
("human", "I love programming."),
|
||||
]
|
||||
chat(...).invoke(messages)
|
||||
```
|
||||
|
||||
chat(...).invoke(messages)
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
AIMessage(content='Je ai une passion pour le programme.\n\nIn
|
||||
French, we use "ai" for masculine subjects and "a" for feminine
|
||||
subjects. Since "programming" is gender-neutral in English, we
|
||||
will go with the masculine "programme".\n\nConfirmation: "J\'aime
|
||||
le programme." is more commonly used. The sentence above is
|
||||
technically accurate, but less commonly used in spoken French as
|
||||
"ai" is used less frequently in everyday speech.',
|
||||
response_metadata={'token_usage': ChatCompletionOutputUsage
|
||||
(completion_tokens=100, prompt_tokens=55, total_tokens=155),
|
||||
'model': '', 'finish_reason': 'length'},
|
||||
id='run-874c24b7-0272-4c99-b259-5d6d7facbc56-0')
|
||||
```python
|
||||
AIMessage(content='Je ai une passion pour le programme.\n\nIn
|
||||
French, we use "ai" for masculine subjects and "a" for feminine
|
||||
subjects. Since "programming" is gender-neutral in English, we
|
||||
will go with the masculine "programme".\n\nConfirmation: "J\'aime
|
||||
le programme." is more commonly used. The sentence above is
|
||||
technically accurate, but less commonly used in spoken French as
|
||||
"ai" is used less frequently in everyday speech.',
|
||||
response_metadata={'token_usage': ChatCompletionOutputUsage
|
||||
(completion_tokens=100, prompt_tokens=55, total_tokens=155),
|
||||
'model': '', 'finish_reason': 'length'},
|
||||
id='run-874c24b7-0272-4c99-b259-5d6d7facbc56-0')
|
||||
```
|
||||
|
||||
Stream:
|
||||
.. code-block:: python
|
||||
```python
|
||||
for chunk in chat.stream(messages):
|
||||
print(chunk)
|
||||
```
|
||||
|
||||
for chunk in chat.stream(messages):
|
||||
print(chunk)
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
content='Je ai une passion pour le programme.\n\nIn French, we use
|
||||
"ai" for masculine subjects and "a" for feminine subjects.
|
||||
Since "programming" is gender-neutral in English,
|
||||
we will go with the masculine "programme".\n\nConfirmation:
|
||||
"J\'aime le programme." is more commonly used. The sentence
|
||||
above is technically accurate, but less commonly used in spoken
|
||||
French as "ai" is used less frequently in everyday speech.'
|
||||
response_metadata={'token_usage': ChatCompletionOutputUsage
|
||||
(completion_tokens=100, prompt_tokens=55, total_tokens=155),
|
||||
'model': '', 'finish_reason': 'length'}
|
||||
id='run-7d7b1967-9612-4f9a-911a-b2b5ca85046a-0'
|
||||
```python
|
||||
content='Je ai une passion pour le programme.\n\nIn French, we use
|
||||
"ai" for masculine subjects and "a" for feminine subjects.
|
||||
Since "programming" is gender-neutral in English,
|
||||
we will go with the masculine "programme".\n\nConfirmation:
|
||||
"J\'aime le programme." is more commonly used. The sentence
|
||||
above is technically accurate, but less commonly used in spoken
|
||||
French as "ai" is used less frequently in everyday speech.'
|
||||
response_metadata={'token_usage': ChatCompletionOutputUsage
|
||||
(completion_tokens=100, prompt_tokens=55, total_tokens=155),
|
||||
'model': '', 'finish_reason': 'length'}
|
||||
id='run-7d7b1967-9612-4f9a-911a-b2b5ca85046a-0'
|
||||
```
|
||||
|
||||
Async:
|
||||
.. code-block:: python
|
||||
```python
|
||||
await chat.ainvoke(messages)
|
||||
```
|
||||
|
||||
await chat.ainvoke(messages)
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
AIMessage(content='Je déaime le programming.\n\nLittérale : Je
|
||||
(j\'aime) déaime (le) programming.\n\nNote: "Programming" in
|
||||
French is "programmation". But here, I used "programming" instead
|
||||
of "programmation" because the user said "I love programming"
|
||||
instead of "I love programming (in French)", which would be
|
||||
"J\'aime la programmation". By translating the sentence
|
||||
literally, I preserved the original meaning of the user\'s
|
||||
sentence.', id='run-fd850318-e299-4735-b4c6-3496dc930b1d-0')
|
||||
```python
|
||||
AIMessage(content='Je déaime le programming.\n\nLittérale : Je
|
||||
(j\'aime) déaime (le) programming.\n\nNote: "Programming" in
|
||||
French is "programmation". But here, I used "programming" instead
|
||||
of "programmation" because the user said "I love programming"
|
||||
instead of "I love programming (in French)", which would be
|
||||
"J\'aime la programmation". By translating the sentence
|
||||
literally, I preserved the original meaning of the user\'s
|
||||
sentence.', id='run-fd850318-e299-4735-b4c6-3496dc930b1d-0')
|
||||
```
|
||||
|
||||
Tool calling:
|
||||
.. code-block:: python
|
||||
```python
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
class GetWeather(BaseModel):
|
||||
'''Get the current weather in a given location'''
|
||||
|
||||
class GetWeather(BaseModel):
|
||||
'''Get the current weather in a given location'''
|
||||
location: str = Field(..., description="The city and state,
|
||||
e.g. San Francisco, CA")
|
||||
|
||||
location: str = Field(..., description="The city and state,
|
||||
e.g. San Francisco, CA")
|
||||
class GetPopulation(BaseModel):
|
||||
'''Get the current population in a given location'''
|
||||
|
||||
class GetPopulation(BaseModel):
|
||||
'''Get the current population in a given location'''
|
||||
location: str = Field(..., description="The city and state,
|
||||
e.g. San Francisco, CA")
|
||||
|
||||
location: str = Field(..., description="The city and state,
|
||||
e.g. San Francisco, CA")
|
||||
chat_with_tools = chat.bind_tools([GetWeather, GetPopulation])
|
||||
ai_msg = chat_with_tools.invoke("Which city is hotter today and
|
||||
which is bigger: LA or NY?")
|
||||
ai_msg.tool_calls
|
||||
```
|
||||
|
||||
chat_with_tools = chat.bind_tools([GetWeather, GetPopulation])
|
||||
ai_msg = chat_with_tools.invoke("Which city is hotter today and
|
||||
which is bigger: LA or NY?")
|
||||
ai_msg.tool_calls
|
||||
|
||||
.. code-block:: python
|
||||
|
||||
[
|
||||
{
|
||||
"name": "GetPopulation",
|
||||
"args": {"location": "Los Angeles, CA"},
|
||||
"id": "0",
|
||||
}
|
||||
]
|
||||
```python
|
||||
[
|
||||
{
|
||||
"name": "GetPopulation",
|
||||
"args": {"location": "Los Angeles, CA"},
|
||||
"id": "0",
|
||||
}
|
||||
]
|
||||
```
|
||||
|
||||
Response metadata
|
||||
.. code-block:: python
|
||||
|
||||
ai_msg = chat.invoke(messages)
|
||||
ai_msg.response_metadata
|
||||
|
||||
.. code-block:: python
|
||||
{
|
||||
"token_usage": ChatCompletionOutputUsage(
|
||||
completion_tokens=100, prompt_tokens=8, total_tokens=108
|
||||
),
|
||||
"model": "",
|
||||
"finish_reason": "length",
|
||||
}
|
||||
```python
|
||||
ai_msg = chat.invoke(messages)
|
||||
ai_msg.response_metadata
|
||||
```
|
||||
|
||||
```python
|
||||
{
|
||||
"token_usage": ChatCompletionOutputUsage(
|
||||
completion_tokens=100, prompt_tokens=8, total_tokens=108
|
||||
),
|
||||
"model": "",
|
||||
"finish_reason": "length",
|
||||
}
|
||||
```
|
||||
""" # noqa: E501
|
||||
|
||||
llm: Any
|
||||
|
||||
@@ -23,19 +23,18 @@ class HuggingFaceEmbeddings(BaseModel, Embeddings):
|
||||
To use, you should have the `sentence_transformers` python package installed.
|
||||
|
||||
Example:
|
||||
.. code-block:: python
|
||||
|
||||
from langchain_huggingface import HuggingFaceEmbeddings
|
||||
|
||||
model_name = "sentence-transformers/all-mpnet-base-v2"
|
||||
model_kwargs = {"device": "cpu"}
|
||||
encode_kwargs = {"normalize_embeddings": False}
|
||||
hf = HuggingFaceEmbeddings(
|
||||
model_name=model_name,
|
||||
model_kwargs=model_kwargs,
|
||||
encode_kwargs=encode_kwargs,
|
||||
)
|
||||
```python
|
||||
from langchain_huggingface import HuggingFaceEmbeddings
|
||||
|
||||
model_name = "sentence-transformers/all-mpnet-base-v2"
|
||||
model_kwargs = {"device": "cpu"}
|
||||
encode_kwargs = {"normalize_embeddings": False}
|
||||
hf = HuggingFaceEmbeddings(
|
||||
model_name=model_name,
|
||||
model_kwargs=model_kwargs,
|
||||
encode_kwargs=encode_kwargs,
|
||||
)
|
||||
```
|
||||
"""
|
||||
|
||||
model_name: str = Field(default=DEFAULT_MODEL_NAME, alias="model")
|
||||
|
||||
@@ -20,17 +20,16 @@ class HuggingFaceEndpointEmbeddings(BaseModel, Embeddings):
|
||||
it as a named parameter to the constructor.
|
||||
|
||||
Example:
|
||||
.. code-block:: python
|
||||
|
||||
from langchain_huggingface import HuggingFaceEndpointEmbeddings
|
||||
|
||||
model = "sentence-transformers/all-mpnet-base-v2"
|
||||
hf = HuggingFaceEndpointEmbeddings(
|
||||
model=model,
|
||||
task="feature-extraction",
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
```python
|
||||
from langchain_huggingface import HuggingFaceEndpointEmbeddings
|
||||
|
||||
model = "sentence-transformers/all-mpnet-base-v2"
|
||||
hf = HuggingFaceEndpointEmbeddings(
|
||||
model=model,
|
||||
task="feature-extraction",
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
```
|
||||
"""
|
||||
|
||||
client: Any = None #: :meta private:
|
||||
|
||||
@@ -34,49 +34,48 @@ class HuggingFaceEndpoint(LLM):
|
||||
or given as a named parameter to the constructor.
|
||||
|
||||
Example:
|
||||
.. code-block:: python
|
||||
```python
|
||||
# Basic Example (no streaming)
|
||||
llm = HuggingFaceEndpoint(
|
||||
endpoint_url="http://localhost:8010/",
|
||||
max_new_tokens=512,
|
||||
top_k=10,
|
||||
top_p=0.95,
|
||||
typical_p=0.95,
|
||||
temperature=0.01,
|
||||
repetition_penalty=1.03,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
|
||||
# Basic Example (no streaming)
|
||||
llm = HuggingFaceEndpoint(
|
||||
endpoint_url="http://localhost:8010/",
|
||||
max_new_tokens=512,
|
||||
top_k=10,
|
||||
top_p=0.95,
|
||||
typical_p=0.95,
|
||||
temperature=0.01,
|
||||
repetition_penalty=1.03,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
# Streaming response example
|
||||
from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
|
||||
|
||||
# Streaming response example
|
||||
from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
|
||||
|
||||
callbacks = [StreamingStdOutCallbackHandler()]
|
||||
llm = HuggingFaceEndpoint(
|
||||
endpoint_url="http://localhost:8010/",
|
||||
max_new_tokens=512,
|
||||
top_k=10,
|
||||
top_p=0.95,
|
||||
typical_p=0.95,
|
||||
temperature=0.01,
|
||||
repetition_penalty=1.03,
|
||||
callbacks=callbacks,
|
||||
streaming=True,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
|
||||
# Basic Example (no streaming) with Mistral-Nemo-Base-2407 model using a third-party provider (Novita).
|
||||
llm = HuggingFaceEndpoint(
|
||||
repo_id="mistralai/Mistral-Nemo-Base-2407",
|
||||
provider="novita",
|
||||
max_new_tokens=100,
|
||||
do_sample=False,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
callbacks = [StreamingStdOutCallbackHandler()]
|
||||
llm = HuggingFaceEndpoint(
|
||||
endpoint_url="http://localhost:8010/",
|
||||
max_new_tokens=512,
|
||||
top_k=10,
|
||||
top_p=0.95,
|
||||
typical_p=0.95,
|
||||
temperature=0.01,
|
||||
repetition_penalty=1.03,
|
||||
callbacks=callbacks,
|
||||
streaming=True,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
|
||||
# Basic Example (no streaming) with Mistral-Nemo-Base-2407 model using a third-party provider (Novita).
|
||||
llm = HuggingFaceEndpoint(
|
||||
repo_id="mistralai/Mistral-Nemo-Base-2407",
|
||||
provider="novita",
|
||||
max_new_tokens=100,
|
||||
do_sample=False,
|
||||
huggingfacehub_api_token="my-api-key",
|
||||
)
|
||||
print(llm.invoke("What is Deep Learning?"))
|
||||
```
|
||||
""" # noqa: E501
|
||||
|
||||
endpoint_url: str | None = None
|
||||
|
||||
@@ -43,31 +43,32 @@ class HuggingFacePipeline(BaseLLM):
|
||||
`summarization` and `translation` for now.
|
||||
|
||||
Example using from_model_id:
|
||||
.. code-block:: python
|
||||
```python
|
||||
from langchain_huggingface import HuggingFacePipeline
|
||||
|
||||
hf = HuggingFacePipeline.from_model_id(
|
||||
model_id="gpt2",
|
||||
task="text-generation",
|
||||
pipeline_kwargs={"max_new_tokens": 10},
|
||||
)
|
||||
```
|
||||
|
||||
from langchain_huggingface import HuggingFacePipeline
|
||||
hf = HuggingFacePipeline.from_model_id(
|
||||
model_id="gpt2",
|
||||
task="text-generation",
|
||||
pipeline_kwargs={"max_new_tokens": 10},
|
||||
)
|
||||
Example passing pipeline in directly:
|
||||
.. code-block:: python
|
||||
|
||||
from langchain_huggingface import HuggingFacePipeline
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
||||
|
||||
model_id = "gpt2"
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
||||
model = AutoModelForCausalLM.from_pretrained(model_id)
|
||||
pipe = pipeline(
|
||||
"text-generation",
|
||||
model=model,
|
||||
tokenizer=tokenizer,
|
||||
max_new_tokens=10,
|
||||
)
|
||||
hf = HuggingFacePipeline(pipeline=pipe)
|
||||
```python
|
||||
from langchain_huggingface import HuggingFacePipeline
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
||||
|
||||
model_id = "gpt2"
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
||||
model = AutoModelForCausalLM.from_pretrained(model_id)
|
||||
pipe = pipeline(
|
||||
"text-generation",
|
||||
model=model,
|
||||
tokenizer=tokenizer,
|
||||
max_new_tokens=10,
|
||||
)
|
||||
hf = HuggingFacePipeline(pipeline=pipe)
|
||||
```
|
||||
"""
|
||||
|
||||
pipeline: Any = None #: :meta private:
|
||||
|
||||
Reference in New Issue
Block a user