Skip to content

Commit 56bbfd9

Browse files
cbornetmdrxy
andauthored
langchain: Add ruff rule RET (#31875)
All auto-fixes See https://docs.astral.sh/ruff/rules/#flake8-return-ret --------- Co-authored-by: Mason Daugherty <[email protected]>
1 parent fceebbb commit 56bbfd9

File tree

91 files changed

+676
-848
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

91 files changed

+676
-848
lines changed

libs/langchain/langchain/__init__.py

Lines changed: 50 additions & 51 deletions
Original file line numberDiff line numberDiff line change
@@ -48,25 +48,25 @@ def __getattr__(name: str) -> Any:
4848
_warn_on_import(name, replacement="langchain.agents.MRKLChain")
4949

5050
return MRKLChain
51-
elif name == "ReActChain":
51+
if name == "ReActChain":
5252
from langchain.agents import ReActChain
5353

5454
_warn_on_import(name, replacement="langchain.agents.ReActChain")
5555

5656
return ReActChain
57-
elif name == "SelfAskWithSearchChain":
57+
if name == "SelfAskWithSearchChain":
5858
from langchain.agents import SelfAskWithSearchChain
5959

6060
_warn_on_import(name, replacement="langchain.agents.SelfAskWithSearchChain")
6161

6262
return SelfAskWithSearchChain
63-
elif name == "ConversationChain":
63+
if name == "ConversationChain":
6464
from langchain.chains import ConversationChain
6565

6666
_warn_on_import(name, replacement="langchain.chains.ConversationChain")
6767

6868
return ConversationChain
69-
elif name == "LLMBashChain":
69+
if name == "LLMBashChain":
7070
msg = (
7171
"This module has been moved to langchain-experimental. "
7272
"For more details: "
@@ -77,153 +77,153 @@ def __getattr__(name: str) -> Any:
7777
)
7878
raise ImportError(msg)
7979

80-
elif name == "LLMChain":
80+
if name == "LLMChain":
8181
from langchain.chains import LLMChain
8282

8383
_warn_on_import(name, replacement="langchain.chains.LLMChain")
8484

8585
return LLMChain
86-
elif name == "LLMCheckerChain":
86+
if name == "LLMCheckerChain":
8787
from langchain.chains import LLMCheckerChain
8888

8989
_warn_on_import(name, replacement="langchain.chains.LLMCheckerChain")
9090

9191
return LLMCheckerChain
92-
elif name == "LLMMathChain":
92+
if name == "LLMMathChain":
9393
from langchain.chains import LLMMathChain
9494

9595
_warn_on_import(name, replacement="langchain.chains.LLMMathChain")
9696

9797
return LLMMathChain
98-
elif name == "QAWithSourcesChain":
98+
if name == "QAWithSourcesChain":
9999
from langchain.chains import QAWithSourcesChain
100100

101101
_warn_on_import(name, replacement="langchain.chains.QAWithSourcesChain")
102102

103103
return QAWithSourcesChain
104-
elif name == "VectorDBQA":
104+
if name == "VectorDBQA":
105105
from langchain.chains import VectorDBQA
106106

107107
_warn_on_import(name, replacement="langchain.chains.VectorDBQA")
108108

109109
return VectorDBQA
110-
elif name == "VectorDBQAWithSourcesChain":
110+
if name == "VectorDBQAWithSourcesChain":
111111
from langchain.chains import VectorDBQAWithSourcesChain
112112

113113
_warn_on_import(name, replacement="langchain.chains.VectorDBQAWithSourcesChain")
114114

115115
return VectorDBQAWithSourcesChain
116-
elif name == "InMemoryDocstore":
116+
if name == "InMemoryDocstore":
117117
from langchain_community.docstore import InMemoryDocstore
118118

119119
_warn_on_import(name, replacement="langchain.docstore.InMemoryDocstore")
120120

121121
return InMemoryDocstore
122-
elif name == "Wikipedia":
122+
if name == "Wikipedia":
123123
from langchain_community.docstore import Wikipedia
124124

125125
_warn_on_import(name, replacement="langchain.docstore.Wikipedia")
126126

127127
return Wikipedia
128-
elif name == "Anthropic":
128+
if name == "Anthropic":
129129
from langchain_community.llms import Anthropic
130130

131131
_warn_on_import(name, replacement="langchain_community.llms.Anthropic")
132132

133133
return Anthropic
134-
elif name == "Banana":
134+
if name == "Banana":
135135
from langchain_community.llms import Banana
136136

137137
_warn_on_import(name, replacement="langchain_community.llms.Banana")
138138

139139
return Banana
140-
elif name == "CerebriumAI":
140+
if name == "CerebriumAI":
141141
from langchain_community.llms import CerebriumAI
142142

143143
_warn_on_import(name, replacement="langchain_community.llms.CerebriumAI")
144144

145145
return CerebriumAI
146-
elif name == "Cohere":
146+
if name == "Cohere":
147147
from langchain_community.llms import Cohere
148148

149149
_warn_on_import(name, replacement="langchain_community.llms.Cohere")
150150

151151
return Cohere
152-
elif name == "ForefrontAI":
152+
if name == "ForefrontAI":
153153
from langchain_community.llms import ForefrontAI
154154

155155
_warn_on_import(name, replacement="langchain_community.llms.ForefrontAI")
156156

157157
return ForefrontAI
158-
elif name == "GooseAI":
158+
if name == "GooseAI":
159159
from langchain_community.llms import GooseAI
160160

161161
_warn_on_import(name, replacement="langchain_community.llms.GooseAI")
162162

163163
return GooseAI
164-
elif name == "HuggingFaceHub":
164+
if name == "HuggingFaceHub":
165165
from langchain_community.llms import HuggingFaceHub
166166

167167
_warn_on_import(name, replacement="langchain_community.llms.HuggingFaceHub")
168168

169169
return HuggingFaceHub
170-
elif name == "HuggingFaceTextGenInference":
170+
if name == "HuggingFaceTextGenInference":
171171
from langchain_community.llms import HuggingFaceTextGenInference
172172

173173
_warn_on_import(
174174
name, replacement="langchain_community.llms.HuggingFaceTextGenInference"
175175
)
176176

177177
return HuggingFaceTextGenInference
178-
elif name == "LlamaCpp":
178+
if name == "LlamaCpp":
179179
from langchain_community.llms import LlamaCpp
180180

181181
_warn_on_import(name, replacement="langchain_community.llms.LlamaCpp")
182182

183183
return LlamaCpp
184-
elif name == "Modal":
184+
if name == "Modal":
185185
from langchain_community.llms import Modal
186186

187187
_warn_on_import(name, replacement="langchain_community.llms.Modal")
188188

189189
return Modal
190-
elif name == "OpenAI":
190+
if name == "OpenAI":
191191
from langchain_community.llms import OpenAI
192192

193193
_warn_on_import(name, replacement="langchain_community.llms.OpenAI")
194194

195195
return OpenAI
196-
elif name == "Petals":
196+
if name == "Petals":
197197
from langchain_community.llms import Petals
198198

199199
_warn_on_import(name, replacement="langchain_community.llms.Petals")
200200

201201
return Petals
202-
elif name == "PipelineAI":
202+
if name == "PipelineAI":
203203
from langchain_community.llms import PipelineAI
204204

205205
_warn_on_import(name, replacement="langchain_community.llms.PipelineAI")
206206

207207
return PipelineAI
208-
elif name == "SagemakerEndpoint":
208+
if name == "SagemakerEndpoint":
209209
from langchain_community.llms import SagemakerEndpoint
210210

211211
_warn_on_import(name, replacement="langchain_community.llms.SagemakerEndpoint")
212212

213213
return SagemakerEndpoint
214-
elif name == "StochasticAI":
214+
if name == "StochasticAI":
215215
from langchain_community.llms import StochasticAI
216216

217217
_warn_on_import(name, replacement="langchain_community.llms.StochasticAI")
218218

219219
return StochasticAI
220-
elif name == "Writer":
220+
if name == "Writer":
221221
from langchain_community.llms import Writer
222222

223223
_warn_on_import(name, replacement="langchain_community.llms.Writer")
224224

225225
return Writer
226-
elif name == "HuggingFacePipeline":
226+
if name == "HuggingFacePipeline":
227227
from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
228228

229229
_warn_on_import(
@@ -232,111 +232,111 @@ def __getattr__(name: str) -> Any:
232232
)
233233

234234
return HuggingFacePipeline
235-
elif name == "FewShotPromptTemplate":
235+
if name == "FewShotPromptTemplate":
236236
from langchain_core.prompts import FewShotPromptTemplate
237237

238238
_warn_on_import(
239239
name, replacement="langchain_core.prompts.FewShotPromptTemplate"
240240
)
241241

242242
return FewShotPromptTemplate
243-
elif name == "Prompt":
243+
if name == "Prompt":
244244
from langchain_core.prompts import PromptTemplate
245245

246246
_warn_on_import(name, replacement="langchain_core.prompts.PromptTemplate")
247247

248248
# it's renamed as prompt template anyways
249249
# this is just for backwards compat
250250
return PromptTemplate
251-
elif name == "PromptTemplate":
251+
if name == "PromptTemplate":
252252
from langchain_core.prompts import PromptTemplate
253253

254254
_warn_on_import(name, replacement="langchain_core.prompts.PromptTemplate")
255255

256256
return PromptTemplate
257-
elif name == "BasePromptTemplate":
257+
if name == "BasePromptTemplate":
258258
from langchain_core.prompts import BasePromptTemplate
259259

260260
_warn_on_import(name, replacement="langchain_core.prompts.BasePromptTemplate")
261261

262262
return BasePromptTemplate
263-
elif name == "ArxivAPIWrapper":
263+
if name == "ArxivAPIWrapper":
264264
from langchain_community.utilities import ArxivAPIWrapper
265265

266266
_warn_on_import(
267267
name, replacement="langchain_community.utilities.ArxivAPIWrapper"
268268
)
269269

270270
return ArxivAPIWrapper
271-
elif name == "GoldenQueryAPIWrapper":
271+
if name == "GoldenQueryAPIWrapper":
272272
from langchain_community.utilities import GoldenQueryAPIWrapper
273273

274274
_warn_on_import(
275275
name, replacement="langchain_community.utilities.GoldenQueryAPIWrapper"
276276
)
277277

278278
return GoldenQueryAPIWrapper
279-
elif name == "GoogleSearchAPIWrapper":
279+
if name == "GoogleSearchAPIWrapper":
280280
from langchain_community.utilities import GoogleSearchAPIWrapper
281281

282282
_warn_on_import(
283283
name, replacement="langchain_community.utilities.GoogleSearchAPIWrapper"
284284
)
285285

286286
return GoogleSearchAPIWrapper
287-
elif name == "GoogleSerperAPIWrapper":
287+
if name == "GoogleSerperAPIWrapper":
288288
from langchain_community.utilities import GoogleSerperAPIWrapper
289289

290290
_warn_on_import(
291291
name, replacement="langchain_community.utilities.GoogleSerperAPIWrapper"
292292
)
293293

294294
return GoogleSerperAPIWrapper
295-
elif name == "PowerBIDataset":
295+
if name == "PowerBIDataset":
296296
from langchain_community.utilities import PowerBIDataset
297297

298298
_warn_on_import(
299299
name, replacement="langchain_community.utilities.PowerBIDataset"
300300
)
301301

302302
return PowerBIDataset
303-
elif name == "SearxSearchWrapper":
303+
if name == "SearxSearchWrapper":
304304
from langchain_community.utilities import SearxSearchWrapper
305305

306306
_warn_on_import(
307307
name, replacement="langchain_community.utilities.SearxSearchWrapper"
308308
)
309309

310310
return SearxSearchWrapper
311-
elif name == "WikipediaAPIWrapper":
311+
if name == "WikipediaAPIWrapper":
312312
from langchain_community.utilities import WikipediaAPIWrapper
313313

314314
_warn_on_import(
315315
name, replacement="langchain_community.utilities.WikipediaAPIWrapper"
316316
)
317317

318318
return WikipediaAPIWrapper
319-
elif name == "WolframAlphaAPIWrapper":
319+
if name == "WolframAlphaAPIWrapper":
320320
from langchain_community.utilities import WolframAlphaAPIWrapper
321321

322322
_warn_on_import(
323323
name, replacement="langchain_community.utilities.WolframAlphaAPIWrapper"
324324
)
325325

326326
return WolframAlphaAPIWrapper
327-
elif name == "SQLDatabase":
327+
if name == "SQLDatabase":
328328
from langchain_community.utilities import SQLDatabase
329329

330330
_warn_on_import(name, replacement="langchain_community.utilities.SQLDatabase")
331331

332332
return SQLDatabase
333-
elif name == "FAISS":
333+
if name == "FAISS":
334334
from langchain_community.vectorstores import FAISS
335335

336336
_warn_on_import(name, replacement="langchain_community.vectorstores.FAISS")
337337

338338
return FAISS
339-
elif name == "ElasticVectorSearch":
339+
if name == "ElasticVectorSearch":
340340
from langchain_community.vectorstores import ElasticVectorSearch
341341

342342
_warn_on_import(
@@ -345,15 +345,15 @@ def __getattr__(name: str) -> Any:
345345

346346
return ElasticVectorSearch
347347
# For backwards compatibility
348-
elif name == "SerpAPIChain" or name == "SerpAPIWrapper":
348+
if name == "SerpAPIChain" or name == "SerpAPIWrapper":
349349
from langchain_community.utilities import SerpAPIWrapper
350350

351351
_warn_on_import(
352352
name, replacement="langchain_community.utilities.SerpAPIWrapper"
353353
)
354354

355355
return SerpAPIWrapper
356-
elif name == "verbose":
356+
if name == "verbose":
357357
from langchain.globals import _verbose
358358

359359
_warn_on_import(
@@ -364,7 +364,7 @@ def __getattr__(name: str) -> Any:
364364
)
365365

366366
return _verbose
367-
elif name == "debug":
367+
if name == "debug":
368368
from langchain.globals import _debug
369369

370370
_warn_on_import(
@@ -375,7 +375,7 @@ def __getattr__(name: str) -> Any:
375375
)
376376

377377
return _debug
378-
elif name == "llm_cache":
378+
if name == "llm_cache":
379379
from langchain.globals import _llm_cache
380380

381381
_warn_on_import(
@@ -386,9 +386,8 @@ def __getattr__(name: str) -> Any:
386386
)
387387

388388
return _llm_cache
389-
else:
390-
msg = f"Could not find: {name}"
391-
raise AttributeError(msg)
389+
msg = f"Could not find: {name}"
390+
raise AttributeError(msg)
392391

393392

394393
__all__ = [

0 commit comments

Comments
 (0)