Refactor OCI code to appease the lint god

Simplify _lint.yml.
Rephrase OCI code base to pass make lint checks.

Author: paxiaatucsdedu

.github/workflows/_lint.yml

@@ -42,17 +42,11 @@ jobs:
           working-directory: ${{ inputs.working-directory }}
           cache-key: lint-with-extras
 
-      - name: Check Poetry File
-        shell: bash
+      - name: Check Poetry configuration
         working-directory: ${{ inputs.working-directory }}
         run: |
           poetry check
-
-      - name: Check lock file
-        shell: bash
-        working-directory: ${{ inputs.working-directory }}
-        run: |
-          poetry lock --check
+          poetry check --lock
 
       - name: Install dependencies
         # Also installs dev/lint/test/typing dependencies, to ensure we have
@@ -65,38 +59,9 @@ jobs:
         # It doesn't matter how you change it, any change will cause a cache-bust.
         working-directory: ${{ inputs.working-directory }}
         run: |
-          poetry install --with lint,typing
-
-      - name: Get .mypy_cache to speed up mypy
-        uses: actions/cache@v4
-        env:
-          SEGMENT_DOWNLOAD_TIMEOUT_MIN: "2"
-        with:
-          path: |
-            ${{ env.WORKDIR }}/.mypy_cache
-          key: mypy-lint-${{ runner.os }}-${{ runner.arch }}-py${{ matrix.python-version }}-${{ inputs.working-directory }}-${{ hashFiles(format('{0}/poetry.lock', inputs.working-directory)) }}
-
-
-      - name: Analysing the code with our lint
-        working-directory: ${{ inputs.working-directory }}
-        run: |
-          make lint_package
-
-      - name: Install unit+integration test dependencies
-        working-directory: ${{ inputs.working-directory }}
-        run: |
-          poetry install --with test,test_integration
-
-      - name: Get .mypy_cache_test to speed up mypy
-        uses: actions/cache@v4
-        env:
-          SEGMENT_DOWNLOAD_TIMEOUT_MIN: "2"
-        with:
-          path: |
-            ${{ env.WORKDIR }}/.mypy_cache_test
-          key: mypy-test-${{ runner.os }}-${{ runner.arch }}-py${{ matrix.python-version }}-${{ inputs.working-directory }}-${{ hashFiles(format('{0}/poetry.lock', inputs.working-directory)) }}
+          poetry install --with lint,typing,test,test_integration
 
-      - name: Analysing the code with our lint
+      - name: Run linting
         working-directory: ${{ inputs.working-directory }}
         run: |
-          make lint_tests
+          make lint

libs/oci/Makefile

@@ -27,14 +27,14 @@ lint_tests: PYTHON_FILES=tests
 lint_tests: MYPY_CACHE=.mypy_cache_test
 
 lint lint_diff lint_package lint_tests:
-	poetry run ruff .
-	poetry run ruff format $(PYTHON_FILES) --diff
-	poetry run ruff --select I $(PYTHON_FILES)
+	poetry run ruff check .
+	poetry run ruff format $(PYTHON_FILES) --check --diff
+	poetry run ruff check --select I $(PYTHON_FILES)
 	mkdir -p $(MYPY_CACHE); poetry run mypy $(PYTHON_FILES) --cache-dir $(MYPY_CACHE)
 
 format format_diff:
 	poetry run ruff format $(PYTHON_FILES)
-	poetry run ruff --select I --fix $(PYTHON_FILES)
+	poetry run ruff check --select I --fix $(PYTHON_FILES)
 
 spell_check:
 	poetry run codespell --toml pyproject.toml

libs/oci/langchain_oci/chat_models/oci_data_science.py

@@ -275,8 +275,7 @@ def validate_openai(cls, values: Any) -> Any:
         """Checks if langchain_openai is installed."""
         if not importlib.util.find_spec("langchain_openai"):
             raise ImportError(
-                "Could not import langchain_openai package. "
-                "Please install it with `pip install langchain_openai`."
+                "Could not import langchain_openai package. Please install it with `pip install langchain_openai`."
             )
         return values
 
@@ -303,9 +302,7 @@ def _default_params(self) -> Dict[str, Any]:
             "stream": self.streaming,
         }
 
-    def _headers(
-        self, is_async: Optional[bool] = False, body: Optional[dict] = None
-    ) -> Dict:
+    def _headers(self, is_async: Optional[bool] = False, body: Optional[dict] = None) -> Dict:
         """Construct and return the headers for a request.
 
         Args:
@@ -357,17 +354,13 @@ def _generate(
                 response = chat.invoke(messages)
         """  # noqa: E501
         if self.streaming:
-            stream_iter = self._stream(
-                messages, stop=stop, run_manager=run_manager, **kwargs
-            )
+            stream_iter = self._stream(messages, stop=stop, run_manager=run_manager, **kwargs)
             return generate_from_stream(stream_iter)
 
         requests_kwargs = kwargs.pop("requests_kwargs", {})
         params = self._invocation_params(stop, **kwargs)
         body = self._construct_json_body(messages, params)
-        res = self.completion_with_retry(
-            data=body, run_manager=run_manager, **requests_kwargs
-        )
+        res = self.completion_with_retry(data=body, run_manager=run_manager, **requests_kwargs)
         return self._process_response(res.json())
 
     def _stream(
@@ -415,9 +408,7 @@ def _stream(
         params = self._invocation_params(stop, **kwargs)
         body = self._construct_json_body(messages, params)  # request json body
 
-        response = self.completion_with_retry(
-            data=body, run_manager=run_manager, stream=True, **requests_kwargs
-        )
+        response = self.completion_with_retry(data=body, run_manager=run_manager, stream=True, **requests_kwargs)
         default_chunk_class = AIMessageChunk
         for line in self._parse_stream(response.iter_lines()):
             chunk = self._handle_sse_line(line, default_chunk_class)
@@ -467,9 +458,7 @@ async def _agenerate(
 
         """  # noqa: E501
         if self.streaming:
-            stream_iter = self._astream(
-                messages, stop=stop, run_manager=run_manager, **kwargs
-            )
+            stream_iter = self._astream(messages, stop=stop, run_manager=run_manager, **kwargs)
             return await agenerate_from_stream(stream_iter)
 
         requests_kwargs = kwargs.pop("requests_kwargs", {})
@@ -593,19 +582,14 @@ def with_structured_output(
                 else JsonOutputParser()
             )
         else:
-            raise ValueError(
-                f"Unrecognized method argument. Expected `json_mode`."
-                f"Received: `{method}`."
-            )
+            raise ValueError(f"Unrecognized method argument. Expected `json_mode`.Received: `{method}`.")
 
         if include_raw:
             parser_assign = RunnablePassthrough.assign(
                 parsed=itemgetter("raw") | output_parser, parsing_error=lambda _: None
             )
             parser_none = RunnablePassthrough.assign(parsed=lambda _: None)
-            parser_with_fallback = parser_assign.with_fallbacks(
-                [parser_none], exception_key="parsing_error"
-            )
+            parser_with_fallback = parser_assign.with_fallbacks([parser_none], exception_key="parsing_error")
             return RunnableMap(raw=llm) | parser_with_fallback
         else:
             return llm | output_parser
@@ -688,9 +672,7 @@ def _process_stream_response(
             if not isinstance(choice, dict):
                 raise TypeError("Endpoint response is not well formed.")
         except (KeyError, IndexError, TypeError) as e:
-            raise ValueError(
-                "Error while formatting response payload for chat model of type"
-            ) from e
+            raise ValueError("Error while formatting response payload for chat model of type") from e
 
         chunk = _convert_delta_to_message_chunk(choice["delta"], default_chunk_cls)
         default_chunk_cls = chunk.__class__
@@ -702,9 +684,7 @@ def _process_stream_response(
         if usage is not None:
             gen_info.update({"usage": usage})
 
-        return ChatGenerationChunk(
-            message=chunk, generation_info=gen_info if gen_info else None
-        )
+        return ChatGenerationChunk(message=chunk, generation_info=gen_info if gen_info else None)
 
     def _process_response(self, response_json: dict) -> ChatResult:
         """Formats response in OpenAI spec.
@@ -729,9 +709,7 @@ def _process_response(self, response_json: dict) -> ChatResult:
             if not isinstance(choices, list):
                 raise TypeError("Endpoint response is not well formed.")
         except (KeyError, TypeError) as e:
-            raise ValueError(
-                "Error while formatting response payload for chat model of type"
-            ) from e
+            raise ValueError("Error while formatting response payload for chat model of type") from e
 
         for choice in choices:
             message = _convert_dict_to_message(choice["message"])